Kevin Fink
commited on
Commit
·
e188ffb
1
Parent(s):
22d6887
deve
Browse files
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 |
|