Spaces:

shorecode
/

gradio-3

Sleeping

Kevin Fink commited on Dec 9, 2024

Commit

a010d6e

1 Parent(s): 2185441

deve

Files changed (1) hide show

app.py CHANGED Viewed

@@ -229,7 +229,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
             train_dataset=train_set['train'],
             eval_dataset=train_set['validation'],
             compute_metrics=compute_metrics,
-            #data_collator=data_collator,
             #processing_class=tokenizer,
         )

             train_dataset=train_set['train'],
             eval_dataset=train_set['validation'],
             compute_metrics=compute_metrics,
+            data_collator=data_collator,
             #processing_class=tokenizer,
         )