From aec3de8bdbe789587a644a41a0d54d673de120bb Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 6 Nov 2023 14:08:27 +0530 Subject: [PATCH] correct pipeline class name (#5652) --- src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py b/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py index 9cc2c8caa7..1f39cc168c 100644 --- a/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py +++ b/src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py @@ -51,9 +51,7 @@ EXAMPLE_DOC_STRING = """ >>> from diffusers import PixArtAlphaPipeline >>> # You can replace the checkpoint id with "PixArt-alpha/PixArt-XL-2-512x512" too. - >>> pipe = StableDiffusionXLPipeline.from_pretrained( - ... "PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16 - ... ) + >>> pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16) >>> # Enable memory optimizations. >>> pipe.enable_model_cpu_offload()