fix style

Signed-off-by: YAO Matrix <[email protected]>

Author: yao-matrix

tests/pipelines/audioldm2/test_audioldm2.py

@@ -45,7 +45,13 @@
     LMSDiscreteScheduler,
     PNDMScheduler,
 )
-from diffusers.utils.testing_utils import backend_empty_cache, enable_full_determinism, is_torch_version, nightly, torch_device
+from diffusers.utils.testing_utils import (
+    backend_empty_cache,
+    enable_full_determinism,
+    is_torch_version,
+    nightly,
+    torch_device,
+)
 
 from ..pipeline_params import TEXT_TO_AUDIO_BATCH_PARAMS, TEXT_TO_AUDIO_PARAMS
 from ..test_pipelines_common import PipelineTesterMixin

tests/pipelines/wan/test_wan.py

@@ -25,6 +25,7 @@
     enable_full_determinism,
     require_torch_accelerator,
     slow,
+    torch_device,
 )
 
 from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_IMAGE_PARAMS, TEXT_TO_IMAGE_PARAMS

tests/quantization/torchao/test_torchao.py

@@ -39,7 +39,6 @@
     numpy_cosine_similarity_distance,
     require_torch,
     require_torch_accelerator,
-    require_torch_gpu,
     require_torchao_version_greater_or_equal,
     slow,
     torch_device,
@@ -342,7 +341,6 @@ def test_device_map(self):
 
                 output = quantized_model(**inputs)[0]
                 output_slice = output.flatten()[-9:].detach().float().cpu().numpy()
-                cos_distance = numpy_cosine_similarity_distance(output_slice, expected_slice)
                 self.assertTrue(numpy_cosine_similarity_distance(output_slice, expected_slice) < 2e-3)
 
             with tempfile.TemporaryDirectory() as offload_folder:

tests/single_file/test_lumina2_transformer.py

@@ -16,8 +16,6 @@
 import gc
 import unittest
 
-import torch
-
 from diffusers import (
     Lumina2Transformer2DModel,
 )

tests/single_file/test_model_flux_transformer_single_file.py

@@ -16,8 +16,6 @@
 import gc
 import unittest
 
-import torch
-
 from diffusers import (
     FluxTransformer2DModel,
 )

tests/single_file/test_sana_transformer.py

@@ -1,8 +1,6 @@
 import gc
 import unittest
 
-import torch
-
 from diffusers import (
     SanaTransformer2DModel,
 )