Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8ed08e4270fdcb889c3a62e016328d7f14171d3c"
Unverified Commit e9a3911b authored by hlky's avatar hlky Committed by GitHub
Browse files

Fix checkpoint in CogView3PlusPipeline example (#10211)

parent 7186bb45
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py
View file @ e9a3911b
...@@ -38,7 +38,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -38,7 +38,7 @@ EXAMPLE_DOC_STRING = """
>>> import torch >>> import torch
>>> from diffusers import CogView3PlusPipeline >>> 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") >>> pipe.to("cuda")
>>> prompt = "A photo of an astronaut riding a horse on mars" >>> prompt = "A photo of an astronaut riding a horse on mars"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment