Skip to content

Commit

Permalink
fix(hf): pass fake eval dataset since it is required
Browse files Browse the repository at this point in the history
  • Loading branch information
shcheklein committed Dec 18, 2024
1 parent a15bb31 commit fc79ef4
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/dvclive/huggingface.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,10 @@ def on_train_end(
):
if self._log_model is True and state.is_world_process_zero:
fake_trainer = Trainer(
args=args, model=kwargs.get("model"), tokenizer=kwargs.get("tokenizer")
args=args,
model=kwargs.get("model"),
tokenizer=kwargs.get("tokenizer"),
eval_dataset=["fake"],
)
name = "best" if args.load_best_model_at_end else "last"
output_dir = os.path.join(args.output_dir, name)
Expand Down

0 comments on commit fc79ef4

Please sign in to comment.