support diffusers ms models

2025-09-08 16:32:50 +08:00
parent 0ae0bd7904
commit 495c3fcd8a
6 changed files with 47 additions and 7 deletions
--- a/patch.py
+++ b/patch.py
@@ -22,7 +22,7 @@ try:
  if torch.npu.is_available() and not torch.cuda.is_available():
      enable_cuda_to_npu_shim()
 except: 
-  print("exception")
+  print("no npu. use native cuda")

 # 1) 可选：如果你的权重来自 lightning 的 ckpt，放行其类（仅在可信来源时）
 try: