diff --git a/comfy/model_management.py b/comfy/model_management.py
index 6093ef63b..49eaf7948 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -96,7 +96,7 @@ except:
     npu_available = False
 
 try:
-    import torch_mlu
+    import torch_mlu  # noqa: F401
     _ = torch.mlu.device_count()
     mlu_available = torch.mlu.is_available()
 except: