diff --git a/comfy/model_management.py b/comfy/model_management.py index 9d39be7b2..a9ac7f590 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