Update run.py

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 
 )
 
@@ -62,4 +62,4 @@ for idx, prompt in enumerate(prompt_list):
     revised_prompt = output.revised_prompts
     images = output.images
     images[0].save(f"../tests/hf_output{idx+1}.png")
-    print(f"Prompt {idx+1} revised: {revised_prompt}")
\ No newline at end of file
+    print(f"Prompt {idx+1} revised: {revised_prompt}")