remove pipeline hooks

2026-10-09 00:09:45 +00:00 · 2024-12-23 13:58:37 +01:00
parent 86e1d1888a
commit 5261bed284
1 changed files with 4 additions and 0 deletions
@@ -813,6 +813,10 @@ 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

            prompt_filename = string_to_filename(prompt)[:25]
            artifacts = {