From 48c2d39e4c9cc9f15025441971b03334101e8781 Mon Sep 17 00:00:00 2001 From: hlky Date: Fri, 13 Dec 2024 11:06:24 +0000 Subject: [PATCH] Fix checkpoint in CogView3PlusPipeline example --- 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 64fff61d2c32..8bed88c275cf 100644 --- a/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py +++ b/src/diffusers/pipelines/cogview3/pipeline_cogview3plus.py @@ -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"