baidu
/

ERNIE-Image-Turbo

ErnieImagePipeline

Model card Files Files and versions

HsiaX commited on Apr 9

Commit

ef10fda

·

verified ·

1 Parent(s): fb3e9bd

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -108,7 +108,7 @@ torch.backends.cudnn.benchmark = False
 # 加载 pipeline
 pipe = ErnieImagePipeline.from_pretrained(
-    "baidu/ERNIE-Image",
     torch_dtype=torch.bfloat16,
 )
 pipe = pipe.to("cuda")

 # 加载 pipeline
 pipe = ErnieImagePipeline.from_pretrained(
+    "baidu/ERNIE-Image-Turbo",
     torch_dtype=torch.bfloat16,
 )
 pipe = pipe.to("cuda")