Skip to content

Commit 657d544

Browse files
committed
fix(wandb): pass fake dataset to avoid exception in trainer (see #34455)
1 parent 33eef99 commit 657d544

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/transformers/integrations/integration_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -918,7 +918,7 @@ def on_train_end(self, args, state, control, model=None, tokenizer=None, **kwarg
918918
if self._log_model.is_enabled and self._initialized and state.is_world_process_zero:
919919
from ..trainer import Trainer
920920

921-
fake_trainer = Trainer(args=args, model=model, processing_class=tokenizer)
921+
fake_trainer = Trainer(args=args, model=model, processing_class=tokenizer, eval_dataset=["fake"])
922922
with tempfile.TemporaryDirectory() as temp_dir:
923923
fake_trainer.save_model(temp_dir)
924924
metadata = (

0 commit comments

Comments
 (0)