From 35740259de2798cf55098c231b7dab19f15e14da Mon Sep 17 00:00:00 2001
From: zhoufan2956 <78578838+zhoufan2956@users.noreply.github.com>
Date: Wed, 12 Feb 2025 19:48:11 +0800
Subject: [PATCH] mix_ascend_bf16_infer_err (#6794)

---
 comfy/model_management.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index 29cd43b51..cbf4c4ea6 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -1082,6 +1082,9 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
 
     if is_intel_xpu():
         return True
+    
+    if is_ascend_npu():
+        return True
 
     props = torch.cuda.get_device_properties(device)
     if props.major >= 8: