Skip to content

Commit d09cf24

Browse files
remove skipped test
1 parent b9dd0cb commit d09cf24

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

tests/pipelines/cosmos/test_cosmos2_5_transfer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -384,3 +384,4 @@ def test_save_load_optional_components(self, expected_max_difference=1e-4):
384384
)
385385
def test_encode_prompt_works_in_isolation(self):
386386
pass
387+

tests/pipelines/test_pipelines_common.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2405,7 +2405,11 @@ def test_pipeline_level_group_offloading_sanity_checks(self):
24052405
if name not in [exclude_module_name] and isinstance(component, torch.nn.Module):
24062406
# `component.device` prints the `onload_device` type. We should probably override the
24072407
# `device` property in `ModelMixin`.
2408-
component_device = next(component.parameters())[0].device
2408+
# Skip modules with no parameters (e.g., dummy safety checkers with only buffers)
2409+
params = list(component.parameters())
2410+
if not params:
2411+
continue
2412+
component_device = params[0].device
24092413
self.assertTrue(torch.device(component_device).type == torch.device(offload_device).type)
24102414

24112415
@require_torch_accelerator

0 commit comments

Comments
 (0)