AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-23 09:09:01 +03:00
Code Activity

[tests] fix group offloading with disk tests (#13491)

Browse Source 
fix group offloading with disk tests
This commit is contained in:
Sayak Paul
2026-04-22 11:41:37 -03:00
committed by GitHub
parent 50987b1290
commit 7c88e5fe21
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/testing_utils/memory.py
View File

@@ -22,10 +22,10 @@ import pytest
import torch
from accelerate.utils.modeling import compute_module_sizes
from diffusers.utils.testing_utils import _check_safetensors_serialization
from diffusers.utils.torch_utils import get_torch_cuda_device_capability
from ...testing_utils import (
_check_safetensors_serialization,
assert_tensors_close,
backend_empty_cache,
backend_max_memory_allocated,
@@ -361,6 +361,9 @@ class GroupOffloadTesterMixin:
offload_to_disk_path=tmpdir,
offload_type=offload_type,
num_blocks_per_group=num_blocks_per_group,
block_modules=model._group_offload_block_modules
if hasattr(model, "_group_offload_block_modules")
else None,
)
if not is_correct:
if extra_files: