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",
|