ovedrive commited on
Commit
7ba4a05
·
verified ·
1 Parent(s): 26398f5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -44,7 +44,7 @@ import torch
44
  from diffusers import ErnieImagePipeline
45
 
46
  pipe = ErnieImagePipeline.from_pretrained(
47
- "Baidu/ERNIE-Image",
48
  torch_dtype=torch.bfloat16,
49
  ).to("cuda")
50
 
 
44
  from diffusers import ErnieImagePipeline
45
 
46
  pipe = ErnieImagePipeline.from_pretrained(
47
+ "ovedrive/ERNIE-Image-nf4",
48
  torch_dtype=torch.bfloat16,
49
  ).to("cuda")
50