GenAICoder commited on
Commit
fe308ac
·
verified ·
1 Parent(s): bde82a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def get_conversational_chain(retriever):
98
  #llm = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True, token=access_token)
99
  #llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
100
  llm = HuggingFacePipeline.from_model_id(
101
- model_id="Phi-3-mini-128k-instruct",
102
  task="text-generation",
103
  pipeline_kwargs={"max_new_tokens": 10})
104
 
 
98
  #llm = AutoModelForCausalLM.from_pretrained("microsoft/Phi-3-mini-128k-instruct", trust_remote_code=True, token=access_token)
99
  #llm = pipeline("text-generation", model="google/gemma-1.1-2b-it")
100
  llm = HuggingFacePipeline.from_model_id(
101
+ model_id="gpt2",
102
  task="text-generation",
103
  pipeline_kwargs={"max_new_tokens": 10})
104