alexkueck commited on
Commit
b7daee2
·
1 Parent(s): 20d05fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -22,6 +22,7 @@ from presets import *
22
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
23
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
24
  base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
 
25
  tokenizer,model,device = load_tokenizer_and_model(base_model,False)
26
 
27
 
 
22
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
23
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
24
  base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
25
+ #base_model = "lmsys/vicuna-13b-v1.3"
26
  tokenizer,model,device = load_tokenizer_and_model(base_model,False)
27
 
28