Fix checkpoint in CogView3PlusPipeline example (#10211)

e9a3911b · hlky · GitHub · 7186bb45 · e9a3911b
Unverified Commit e9a3911b authored Dec 16, 2024 by hlky Committed by GitHub Dec 16, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py +1 -1

No files found.
--- a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py
+++ b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py
@@ -38,7 +38,7 @@ EXAMPLE_DOC_STRING = """
        >>> 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"