Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,7 +10,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
|
| 10 |
# -------------------- Config -------------------- #
|
| 11 |
|
| 12 |
EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
|
| 13 |
-
LLM_MODEL_NAME = "google/gemma-2b-it" # you can change this later
|
|
|
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
# -------------------- Model loaders (cached) -------------------- #
|
|
|
|
| 10 |
# -------------------- Config -------------------- #
|
| 11 |
|
| 12 |
EMBEDDING_MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
|
| 13 |
+
#LLM_MODEL_NAME = "google/gemma-2b-it" # you can change this later
|
| 14 |
+
LLM_MODEL_NAME = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
|
| 15 |
+
|
| 16 |
|
| 17 |
|
| 18 |
# -------------------- Model loaders (cached) -------------------- #
|