alexkueck commited on
Commit
6759744
·
1 Parent(s): 806c651

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,7 +44,7 @@ xxx = os.getenv("TOKEN")
44
  login(token=xxx)
45
  #Modelle und Tokenizer
46
  model_name = "alexkueck/LIFineTuned"
47
-
48
  #Alternativ mit beliebigen Modellen:
49
  base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
50
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
@@ -87,7 +87,7 @@ lm_datasets = tokenized_datasets.map(
87
  ####################################################
88
 
89
  #Training Args
90
- #model_name = base_model.split("/")[-1]
91
  training_args = TrainingArguments(
92
  f"{model_name}/model/",
93
  evaluation_strategy = "epoch",
 
44
  login(token=xxx)
45
  #Modelle und Tokenizer
46
  model_name = "alexkueck/LIFineTuned"
47
+ print(f"{model_name}/model/")
48
  #Alternativ mit beliebigen Modellen:
49
  base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
50
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
 
87
  ####################################################
88
 
89
  #Training Args
90
+ #model_name = base_model.split("/")[-1]
91
  training_args = TrainingArguments(
92
  f"{model_name}/model/",
93
  evaluation_strategy = "epoch",