From b60bf2dacc8e46e40937e2ecab77ca42515ec9bb Mon Sep 17 00:00:00 2001 From: DN6 Date: Thu, 19 Jun 2025 21:31:44 +0530 Subject: [PATCH] update --- src/diffusers/hooks/group_offloading.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/hooks/group_offloading.py b/src/diffusers/hooks/group_offloading.py index 5dce3fbf35..dfe312add5 100644 --- a/src/diffusers/hooks/group_offloading.py +++ b/src/diffusers/hooks/group_offloading.py @@ -92,8 +92,8 @@ class ModuleGroup: 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()} - keys_str = "_".join(sorted(self.key_to_tensor.keys())) - self._disk_offload_group_id = hashlib.md5(keys_str.encode()).hexdigest()[:8] + group_id_key = "_".join(sorted([param_name for param_name, _ in module.named_parameters()])) + self._disk_offload_group_id = hashlib.md5(group_id_key.encode()).hexdigest()[:8] self.cpu_param_dict = {} else: