print

Author: sayakpaul

src/diffusers/pipelines/pipeline_utils.py

@@ -986,6 +986,7 @@ def remove_all_hooks(self):
         print("Within remove_all_hooks().")
         for _, model in self.components.items():
             if isinstance(model, torch.nn.Module) and hasattr(model, "_hf_hook"):
+                print(f"{model.__class__.__name__=}")
                 accelerate.hooks.remove_hook_from_module(model, recurse=True)
         self._all_hooks = []