better check

Author: a-r-r-o-w

src/diffusers/models/modeling_utils.py

@@ -86,9 +86,11 @@ def get_parameter_device(parameter: torch.nn.Module) -> torch.device:
     try:
         if hasattr(parameter, "_diffusers_hook"):
             for submodule in parameter.modules():
-                if hasattr(submodule, "_diffusers_hook"):
-                    registry = parameter._diffusers_hook
-                    hook = registry.get_hook("group_offloading")
+                if not hasattr(submodule, "_diffusers_hook"):
+                    continue
+                registry = parameter._diffusers_hook
+                hook = registry.get_hook("group_offloading")
+                if hook is not None:
                     return hook.group.onload_device
 
         parameters_and_buffers = itertools.chain(parameter.parameters(), parameter.buffers())