diff --git a/comfy/model_management.py b/comfy/model_management.py index cd8772d3a..9ccd0707b 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -131,6 +131,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