Skip to content

Commit

Permalink
Fix checkpoint in CogView3PlusPipeline example (#10211)
Browse files Browse the repository at this point in the history
  • Loading branch information
hlky authored Dec 16, 2024
1 parent 7186bb4 commit e9a3911
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@
>>> import torch
>>> from diffusers import CogView3PlusPipeline
>>> pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3Plus-3B", torch_dtype=torch.bfloat16)
>>> pipe = CogView3PlusPipeline.from_pretrained("THUDM/CogView3-Plus-3B", torch_dtype=torch.bfloat16)
>>> pipe.to("cuda")
>>> prompt = "A photo of an astronaut riding a horse on mars"
Expand Down

0 comments on commit e9a3911

Please sign in to comment.