diff --git a/comfy/model_management.py b/comfy/model_management.py index c5b817b62643..8557612793c7 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -121,6 +121,7 @@ def get_supported_float8_types(): 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