Kevin Fink commited on
Commit
a67d62e
·
1 Parent(s): c789f89
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -224,7 +224,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
224
  train_dataset=train_set['train'],
225
  eval_dataset=train_set['validation'],
226
  compute_metrics=compute_metrics,
227
- data_collator=data_collator,
228
  #processing_class=tokenizer,
229
  )
230
 
 
224
  train_dataset=train_set['train'],
225
  eval_dataset=train_set['validation'],
226
  compute_metrics=compute_metrics,
227
+ #data_collator=data_collator,
228
  #processing_class=tokenizer,
229
  )
230