Update app.py

app.py

@@ -79,9 +79,9 @@ login(token=os.environ["HF_ACCESS_TOKEN"])   #for read access!!!!
 #Modelle und Tokenizer
 
 #Alternativ mit beliebigen Modellen:
-#base_model = "project-baize/baize-v2-7b"  #load_8bit = False (in load_tokenizer_and_model)
+base_model = "project-baize/baize-v2-7b"  #load_8bit = False (in load_tokenizer_and_model)
 #base_model = "TheBloke/airoboros-13B-HF"  #load_8bit = False (in load_tokenizer_and_model)
-base_model = "EleutherAI/gpt-neo-1.3B"    #load_8bit = False (in load_tokenizer_and_model)
+#base_model = "EleutherAI/gpt-neo-1.3B"    #load_8bit = False (in load_tokenizer_and_model)
 #base_model = "TheBloke/airoboros-13B-HF"   #load_8bit = True
 
 #Tokenizer und Model laden
@@ -140,7 +140,7 @@ print ("training args")
 batch_size = 2
 
 training_args = TrainingArguments(
-    output_dir="alexkueck/li-tis-tuned-1",
+    output_dir="alexkueck/li-tis-tuned-2",
     overwrite_output_dir = 'True',
     per_device_train_batch_size=batch_size,  #batch_size = 2 for full training
     per_device_eval_batch_size=batch_size,
@@ -221,8 +221,8 @@ print("Done Eval")
 #Push to Hub
 print("push to hub")
 login(token=os.environ["HF_WRITE_TOKEN"]) 
-#trainer.push_to_hub("alexkueck/li-tis-tuned-1")
-tokenizer.push_to_hub("alexkueck/li-tis-tuned-1")
+trainer.push_to_hub("alexkueck/li-tis-tuned-2")
+tokenizer.push_to_hub("alexkueck/li-tis-tuned-2")
 print("done")