diff --git a/comfy/model_management.py b/comfy/model_management.py index 928282092..906350a47 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -123,6 +123,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