captain-awesome
commited on
Commit
•
94701f4
1
Parent(s):
a9a1953
Update app.py
Browse files
app.py
CHANGED
@@ -82,7 +82,7 @@ def get_response(user_input):
|
|
82 |
# lib="avx2", # for CPU
|
83 |
# )
|
84 |
|
85 |
-
llm_model == "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
86 |
llm = HuggingFaceHub(
|
87 |
repo_id=llm_model,
|
88 |
model_kwargs={"temperature": temperature, "max_new_tokens": 250, "top_k": top_k}
|
|
|
82 |
# lib="avx2", # for CPU
|
83 |
# )
|
84 |
|
85 |
+
llm_model == "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
86 |
llm = HuggingFaceHub(
|
87 |
repo_id=llm_model,
|
88 |
model_kwargs={"temperature": temperature, "max_new_tokens": 250, "top_k": top_k}
|