update

Author: DN6

src/diffusers/hooks/group_offloading.py

@@ -82,17 +82,21 @@ def __init__(
 
         if self.offload_to_disk_path:
             all_tensors = []
+            param_names = []
             for module in self.modules:
                 all_tensors.extend(list(module.parameters()))
                 all_tensors.extend(list(module.buffers()))
+
+                param_names.extend([param_name for param_name, _ in module.named_parameters()])
+
             all_tensors.extend(self.parameters)
             all_tensors.extend(self.buffers)
             all_tensors = list(dict.fromkeys(all_tensors))  # Remove duplicates
 
             self.tensor_to_key = {tensor: f"tensor_{i}" for i, tensor in enumerate(all_tensors)}
             self.key_to_tensor = {v: k for k, v in self.tensor_to_key.items()}
 
-            group_id_key = "_".join(sorted([param_name for param_name, _ in module.named_parameters()]))
+            group_id_key = "_".join(sorted(param_names))
             self._disk_offload_group_id = hashlib.md5(group_id_key.encode()).hexdigest()[:8]
 
             self.cpu_param_dict = {}