Skip to content

Commit

Permalink
quality
Browse files Browse the repository at this point in the history
  • Loading branch information
sayakpaul committed Jan 16, 2025
1 parent 60e2c58 commit bbd817d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion finetrainers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -964,7 +964,7 @@ def validate(self, step: int, final_validation: bool = False) -> None:
if video_artifacts:
artifact_log_dict["videos"] = video_artifacts
tracker.log({tracker_key: artifact_log_dict}, step=step)

if self.args.push_to_hub and final_validation:
video_filenames = list(prompts_to_filenames.values())
prompts = list(prompts_to_filenames.keys())
Expand Down

0 comments on commit bbd817d

Please sign in to comment.