Skip to content

Commit 6938524

Browse files
authored
fix(dvclive): pass fake dataset to avoid exception in trainer init (#34455)
fix(dvclive): pass fake dataset to avoid exception in trainer
1 parent 7bbc624 commit 6938524

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/transformers/integrations/integration_utils.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -2125,7 +2125,12 @@ def on_train_end(self, args, state, control, **kwargs):
21252125
from transformers.trainer import Trainer
21262126

21272127
if self._log_model is True:
2128-
fake_trainer = Trainer(args=args, model=kwargs.get("model"), processing_class=kwargs.get("tokenizer"))
2128+
fake_trainer = Trainer(
2129+
args=args,
2130+
model=kwargs.get("model"),
2131+
processing_class=kwargs.get("tokenizer"),
2132+
eval_dataset=["fake"],
2133+
)
21292134
name = "best" if args.load_best_model_at_end else "last"
21302135
output_dir = os.path.join(args.output_dir, name)
21312136
fake_trainer.save_model(output_dir)

0 commit comments

Comments
 (0)