Update app.py
Browse files
app.py
CHANGED
|
@@ -196,8 +196,7 @@ def run_training(hf_token, model_name, new_repo_name, lora_r, lora_alpha, lora_d
|
|
| 196 |
logging_steps=5,
|
| 197 |
save_strategy="steps",
|
| 198 |
save_steps=save_steps_val,
|
| 199 |
-
report_to="none"
|
| 200 |
-
fp16=torch.cuda.is_available()
|
| 201 |
)
|
| 202 |
|
| 203 |
processed_dataset = IterableDataset.from_generator(processed_samples_generator)
|
|
|
|
| 196 |
logging_steps=5,
|
| 197 |
save_strategy="steps",
|
| 198 |
save_steps=save_steps_val,
|
| 199 |
+
report_to="none"
|
|
|
|
| 200 |
)
|
| 201 |
|
| 202 |
processed_dataset = IterableDataset.from_generator(processed_samples_generator)
|