PetrovDE commited on
Commit
4700841
1 Parent(s): e39d37a

change app to low model 15m

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ class Prompter(object):
35
 
36
 
37
  load_8bit = False # for Colab
38
- base_model = "baffo32/decapoda-research-llama-7B-hf"
39
- lora_weights = "tiedong/goat-lora-7b"
40
  share_gradio = True
41
 
42
  if torch.cuda.is_available():
 
35
 
36
 
37
  load_8bit = False # for Colab
38
+ base_model = "nickypro/tinyllama-15M"
39
+ lora_weights = "./chkp"
40
  share_gradio = True
41
 
42
  if torch.cuda.is_available():