alexkueck commited on
Commit
17a3582
1 Parent(s): b54983b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_
20
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
21
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
22
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
23
- tokenizer,model,device = load_tokenizer_and_model(base_model,True)
24
 
25
 
26
  ########################################################################
 
20
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
21
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
22
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
23
+ tokenizer,model,device = load_tokenizer_and_model(base_model,False)
24
 
25
 
26
  ########################################################################