Avinashstat commited on
Commit
d67a424
·
verified ·
1 Parent(s): 9aa38c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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) -------------------- #