handaru2002 commited on
Commit
f5f26d0
·
verified ·
1 Parent(s): 7698adc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -30,7 +30,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
30
  # SMALL LLM (POST-EDITOR)
31
  # ============================================================
32
 
33
- LLM = "microsoft/phi-2"
 
34
 
35
  # =========================
36
  # 1. LOAD TOKENIZER
 
30
  # SMALL LLM (POST-EDITOR)
31
  # ============================================================
32
 
33
+
34
+ LLM = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
35
 
36
  # =========================
37
  # 1. LOAD TOKENIZER