Kevin Fink commited on
Commit
e188ffb
·
1 Parent(s): 22d6887
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -151,7 +151,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
151
  train_dataset=train_dataset,
152
  eval_dataset=saved_test_dataset,
153
  #compute_metrics=compute_metrics,
154
- data_collator=data_collator,
155
  #processing_class=tokenizer,
156
  )
157
 
 
151
  train_dataset=train_dataset,
152
  eval_dataset=saved_test_dataset,
153
  #compute_metrics=compute_metrics,
154
+ #data_collator=data_collator,
155
  #processing_class=tokenizer,
156
  )
157