diff --git a/comfy/model_management.py b/comfy/model_management.py index e5de4a5b5..e0a96843c 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -122,6 +122,7 @@ except: try: import torch_npu # noqa: F401 _ = torch.npu.device_count() + torch_npu.npu.set_compile_mode(jit_compile = False) npu_available = torch.npu.is_available() except: npu_available = False