format fix with ruff cmd

Author: Pei Zhang

src/diffusers/models/attention_processor.py

@@ -21,7 +21,7 @@
 
 from ..image_processor import IPAdapterMaskProcessor
 from ..utils import deprecate, is_torch_xla_available, logging
-from ..utils.import_utils import is_torch_npu_available, is_xformers_available, is_torch_xla_version
+from ..utils.import_utils import is_torch_npu_available, is_torch_xla_version, is_xformers_available
 from ..utils.torch_utils import is_torch_version, maybe_allow_in_graph
 
 
@@ -39,8 +39,8 @@
 if is_torch_xla_available():
     # flash attention pallas kernel is introduced in the torch_xla 2.3 release.
     if is_torch_xla_version(">", "2.2"):
-        from torch_xla.runtime import is_spmd
         from torch_xla.experimental.custom_kernel import flash_attention
+        from torch_xla.runtime import is_spmd
     XLA_AVAILABLE = True
 else:
     XLA_AVAILABLE = False

src/diffusers/models/modeling_utils.py

@@ -226,13 +226,13 @@ def fn_recursive_set_flash_attention(module: torch.nn.Module):
                 fn_recursive_set_flash_attention(module)
 
     def enable_xla_flash_attention(self, partition_spec: Optional[Callable] = None):
-        r""" 
+        r"""
         Enable the flash attention pallals kernel for torch_xla.
         """
         self.set_use_xla_flash_attention(True, partition_spec)
 
     def disable_xla_flash_attention(self):
-        r""" 
+        r"""
         Disable the flash attention pallals kernel for torch_xla.
         """
         self.set_use_xla_flash_attention(False)