Spaces:

bstraehle
/

sft

Running

bstraehle commited on Aug 3

Commit

5a35f8f

•

1 Parent(s): 1afcb19

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,7 +79,7 @@ def fine_tune_model(base_model_name, dataset_name):
     training_args = Seq2SeqTrainingArguments(
         output_dir="./output",
         logging_dir="./logging",
-        num_train_epochs=1
         max_steps=2, ###
         #per_device_train_batch_size=16,
         #per_device_eval_batch_size=64,

     training_args = Seq2SeqTrainingArguments(
         output_dir="./output",
         logging_dir="./logging",
+        num_train_epochs=1,
         max_steps=2, ###
         #per_device_train_batch_size=16,
         #per_device_eval_batch_size=64,