AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

correct pipeline class name (#5652)

Browse Source
This commit is contained in:
Sayak Paul
2023-11-06 14:08:27 +05:30
committed by GitHub
parent d61889fc17
commit aec3de8bdb
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/pipelines/pixart_alpha/pipeline_pixart_alpha.py
View File

@@ -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()