alexkueck commited on
Commit
50daa38
·
1 Parent(s): 1a07f51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,9 +57,9 @@ login(token=xxx)
57
  #Modelle und Tokenizer
58
 
59
  #Alternativ mit beliebigen Modellen:
60
- base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
61
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
62
- #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
63
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
64
  tokenizer,model,device = load_tokenizer_and_model(base_model, False)
65
 
 
57
  #Modelle und Tokenizer
58
 
59
  #Alternativ mit beliebigen Modellen:
60
+ #base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
61
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
62
+ base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
63
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
64
  tokenizer,model,device = load_tokenizer_and_model(base_model, False)
65