Skip to content

Commit ebde934

Browse files
committed
update
1 parent 4854309 commit ebde934

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

src/diffusers/hooks/group_offloading.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
import hashlib
1615
import os
1716
from contextlib import contextmanager, nullcontext
1817
from typing import Dict, List, Optional, Set, Tuple, Union
@@ -192,11 +191,6 @@ def _onload_from_memory(self, current_stream):
192191

193192
@torch.compiler.disable()
194193
def onload_(self):
195-
# Generate disk offload group ID if needed and not already set
196-
if self.offload_to_disk_path and self._disk_offload_group_id is None:
197-
keys_str = "_".join(sorted(self.key_to_tensor.keys()))
198-
self._disk_offload_group_id = hashlib.md5(keys_str.encode()).hexdigest()[:8]
199-
200194
torch_accelerator_module = (
201195
getattr(torch, torch.accelerator.current_accelerator().type)
202196
if hasattr(torch, "accelerator")

0 commit comments

Comments
 (0)