fix style

yao-matrix · yao-matrix · commit 2ed18b58d1cc · 2025-05-28T06:35:43.000Z
Signed-off-by: Matrix YAO &lt;matrix.yao@intel.com&gt;
diff --git a/tests/hooks/test_group_offloading.py b/tests/hooks/test_group_offloading.py
@@ -22,7 +22,13 @@
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 from diffusers.utils import get_logger
 from diffusers.utils.import_utils import compare_versions
-from diffusers.utils.testing_utils import backend_empty_cache, backend_max_memory_allocated, backend_reset_peak_memory_stats, require_torch_accelerator, require_torch_gpu, torch_device
+from diffusers.utils.testing_utils import (
+    backend_empty_cache,
+    backend_max_memory_allocated,
+    backend_reset_peak_memory_stats,
+    require_torch_accelerator,
+    torch_device,
+)
 
 
 class DummyBlock(torch.nn.Module):
diff --git a/tests/pipelines/test_pipeline_utils.py b/tests/pipelines/test_pipeline_utils.py
@@ -19,7 +19,7 @@
     UNet2DConditionModel,
 )
 from diffusers.pipelines.pipeline_loading_utils import is_safetensors_compatible, variant_compatible_siblings
-from diffusers.utils.testing_utils import require_torch_accelerator, require_torch_gpu, torch_device
+from diffusers.utils.testing_utils import require_torch_accelerator, torch_device
 
 
 class IsSafetensorsCompatibleTests(unittest.TestCase):

Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@`
`19`	`19`	`UNet2DConditionModel,`
`20`	`20`	`)`
`21`	`21`	`from diffusers.pipelines.pipeline_loading_utils import is_safetensors_compatible, variant_compatible_siblings`
`22`		`-from diffusers.utils.testing_utils import require_torch_accelerator, require_torch_gpu, torch_device`
	`22`	`+from diffusers.utils.testing_utils import require_torch_accelerator, torch_device`
`23`	`23`
`24`	`24`
`25`	`25`	`class IsSafetensorsCompatibleTests(unittest.TestCase):`