update comment

This commit is contained in:
Aryan
2024-12-23 14:01:39 +01:00
parent 75a854f9df
commit 78712f60a5
2 changed files with 5 additions and 2 deletions
+1 -1
View File
@@ -813,7 +813,7 @@ class Trainer:
num_videos_per_prompt=self.args.num_validation_videos_per_prompt,
generator=self.state.generator,
)
# Remove all hooks that might have been added during pipeline initialization to the models
pipeline.remove_all_hooks()
del pipeline