alexkueck commited on
Commit
d569d83
·
1 Parent(s): 9829890

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -82,10 +82,11 @@ login(token=os.environ["HF_ACCESS_TOKEN"]) #for read access!!!!
82
  #base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
83
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
84
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
85
- base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
 
86
 
87
  #Tokenizer und Model laden
88
- tokenizer,model,device = load_tokenizer_and_model(base_model, True)
89
  #tokenizer.add_special_tokens({'pad_token': '[PAD]'}) #not necessary with fast Toekenizers like GPT2
90
 
91
  #Datensets für Finetuning laden
 
82
  #base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
83
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
84
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
85
+ #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
86
+ base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
87
 
88
  #Tokenizer und Model laden
89
+ tokenizer,model,device = load_tokenizer_and_model(base_model, False)
90
  #tokenizer.add_special_tokens({'pad_token': '[PAD]'}) #not necessary with fast Toekenizers like GPT2
91
 
92
  #Datensets für Finetuning laden