From 50ff3a2a6a32341010de370ae7cf6237a5b1c6b9 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Wed, 26 Feb 2025 20:41:38 -0500 Subject: [PATCH] Ruff. --- comfy/model_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 6093ef63..49eaf794 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: