Update run.py

d1d45737 · raojy · 380f7a99 · d1d45737
Commit d1d45737 authored Apr 14, 2026 by raojy 💬
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

run.py run.py +2 -2

No files found.
--- a/run.py
+++ b/run.py
@@ -22,7 +22,7 @@ torch.backends.cudnn.benchmark = False
 # 加载 pipeline
 # 注意：如果你的 DCU 版本（如某些较老的型号）对 bfloat16 支持不佳，可以尝试换成 torch.float16
 pipe = ErnieImagePipeline.from_pretrained(
-    "/public/home/raojy/project/baidu/ERNIE-Image",
+    "baidu/ERNIE-Image",
    torch_dtype=torch.bfloat16 
 )