fix comment.

Author: a120092009

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
+from .import_utils import is_torch_available, is_torch_npu_available, is_torch_version, is_mlu_available
 
 
 if is_torch_available():
@@ -286,7 +286,7 @@ def get_device():
         return "xpu"
     elif torch.backends.mps.is_available():
         return "mps"
-    elif torch.mlu.is_available():
+    elif is_mlu_available():
         return "mlu"
     else:
         return "cpu"