Skip to content

GitLab

Unverified Commit aec3de8b authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

correct pipeline class name (#5652)

parent d61889fc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py
View file @ aec3de8b
...@@ -51,9 +51,7 @@ EXAMPLE_DOC_STRING = """ ...@@ -51,9 +51,7 @@ EXAMPLE_DOC_STRING = """
>>> from diffusers import PixArtAlphaPipeline >>> from diffusers import PixArtAlphaPipeline
>>> # You can replace the checkpoint id with "PixArt-alpha/PixArt-XL-2-512x512" too. >>> # You can replace the checkpoint id with "PixArt-alpha/PixArt-XL-2-512x512" too.
>>> pipe = StableDiffusionXLPipeline.from_pretrained( >>> pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
... "PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16
... )
>>> # Enable memory optimizations. >>> # Enable memory optimizations.
>>> pipe.enable_model_cpu_offload() >>> pipe.enable_model_cpu_offload()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment