Skip to content

Commit 405f13e

Browse files
hlkysayakpaul
authored andcommitted
Fix checkpoint in CogView3PlusPipeline example (#10211)
1 parent 3d11ced commit 405f13e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
>>> import torch
3939
>>> from diffusers import CogView3PlusPipeline
4040
41-
>>> pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3Plus-3B", torch_dtype=torch.bfloat16)
41+
>>> pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3-Plus-3B", torch_dtype=torch.bfloat16)
4242
>>> pipe.to("cuda")
4343
4444
>>> prompt = "A photo of an astronaut riding a horse on mars"

0 commit comments

Comments
 (0)