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