AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

update

Browse Source
This commit is contained in:
DN6
2025-06-19 21:31:44 +05:30
parent 22e6fc180c
commit b60bf2dacc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/hooks/group_offloading.py
View File

@@ -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: