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