Spaces:
Paused
Paused
Update app.py
Browse files
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,
|
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
|