update

Author: a-r-r-o-w

src/diffusers/hooks/hooks.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 
 import functools
-import gc
 from typing import Any, Dict, Optional, Tuple
 
 import torch
@@ -136,7 +135,10 @@ def __init__(self, module_ref: torch.nn.Module) -> None:
 
     def register_hook(self, hook: ModelHook, name: str) -> None:
         if name in self.hooks.keys():
-            logger.warning(f"Hook with name {name} already exists, replacing it.")
+            raise ValueError(
+                f"Hook with name {name} already exists in the registry. Please use a different name or "
+                f"first remove the existing hook and then add a new one."
+            )
 
         self._module_ref = hook.initialize_hook(self._module_ref)
 
@@ -205,8 +207,6 @@ def remove_hook(self, name: str, recurse: bool = True) -> None:
                 if hasattr(module, "_diffusers_hook"):
                     module._diffusers_hook.remove_hook(name, recurse=False)
 
-        gc.collect()
-
     def reset_stateful_hooks(self, recurse: bool = True) -> None:
         for hook_name in reversed(self._hook_order):
             hook = self.hooks[hook_name]