rename is_mlu_available to is_torch_mlu_available

Author: a120092009

src/diffusers/pipelines/pipeline_utils.py

@@ -62,7 +62,7 @@
     is_accelerate_version,
     is_hpu_available,
     is_torch_npu_available,
-    is_mlu_available,
+    is_torch_mlu_available,
     is_torch_version,
     is_transformers_version,
     logging,

src/diffusers/utils/__init__.py

@@ -81,7 +81,6 @@
     is_google_colab,
     is_hf_hub_version,
     is_hpu_available,
-    is_mlu_available,
     is_inflect_available,
     is_invisible_watermark_available,
     is_k_diffusion_available,
@@ -113,6 +112,7 @@
     is_torch_version,
     is_torch_xla_available,
     is_torch_xla_version,
+    is_torch_mlu_available,
     is_torchao_available,
     is_torchao_version,
     is_torchsde_available,

src/diffusers/utils/import_utils.py

@@ -244,7 +244,7 @@ def is_torch_npu_available():
     return _torch_npu_available
 
 
-def is_mlu_available():
+def is_torch_mlu_available():
     return _torch_mlu_available
 
 

src/diffusers/utils/torch_utils.py

@@ -20,7 +20,7 @@
 from typing import Callable, Dict, List, Optional, Tuple, Union
 
 from . import logging
-from .import_utils import is_torch_available, is_torch_npu_available, is_torch_version, is_mlu_available
+from .import_utils import is_torch_available, is_torch_npu_available, is_torch_version, is_torch_mlu_available
 
 
 if is_torch_available():
@@ -286,7 +286,7 @@ def get_device():
         return "xpu"
     elif torch.backends.mps.is_available():
         return "mps"
-    elif is_mlu_available():
+    elif is_torch_mlu_available():
         return "mlu"
     else:
         return "cpu"