Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -125,6 +125,7 @@ trainer = Trainer(
|
|
125 |
args=training_args,
|
126 |
train_dataset=lm_datasets["train"],
|
127 |
eval_dataset=lm_datasets["test"],
|
|
|
128 |
#compute_metrics=compute_metrics,
|
129 |
)
|
130 |
print ("trainer")
|
|
|
125 |
args=training_args,
|
126 |
train_dataset=lm_datasets["train"],
|
127 |
eval_dataset=lm_datasets["test"],
|
128 |
+
tokenizer=tokenizer,
|
129 |
#compute_metrics=compute_metrics,
|
130 |
)
|
131 |
print ("trainer")
|