From e9a3911b676fa0ec309999fb89fd5fd686495c42 Mon Sep 17 00:00:00 2001 From: hlky Date: Mon, 16 Dec 2024 19:31:22 +0000 Subject: [PATCH] Fix checkpoint in CogView3PlusPipeline example (#10211) --- src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py index 64fff61d2c..8bed88c275 100644 --- 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"