Ksjsjjdj commited on
Commit
c5fcfaf
·
verified ·
1 Parent(s): a080efa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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)