Skip to content

Commit aec3de8

Browse files
authored
correct pipeline class name (#5652)
1 parent d61889f commit aec3de8

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,9 +51,7 @@
5151
>>> from diffusers import PixArtAlphaPipeline
5252
5353
>>> # You can replace the checkpoint id with "PixArt-alpha/PixArt-XL-2-512x512" too.
54-
>>> pipe = StableDiffusionXLPipeline.from_pretrained(
55-
... "PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16
56-
... )
54+
>>> pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
5755
>>> # Enable memory optimizations.
5856
>>> pipe.enable_model_cpu_offload()
5957

0 commit comments

Comments
 (0)