Spaces:
Running
Running
Update app.py
Browse files
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="
|
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 |
|