Spaces:

shorecode
/

gradio-3

Sleeping

Kevin Fink commited on Dec 9, 2024

Commit

c789f89

1 Parent(s): 730452f

deve

Files changed (1) hide show

app.py CHANGED Viewed

@@ -224,7 +224,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,
-            sdata_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,
         )