PixArt-alpha
/

PixArt-XL-2-1024-MS

PixArtAlphaPipeline

Model card Files Files and versions

yujincheng08 commited on Nov 4, 2023

Commit

196525a

•

1 Parent(s): 8e10ed6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -6,7 +6,7 @@ Usage:
 ```python
 from diffusers import PixArtAlphaPipeline
 import torch
-pipe = PixArtAlphaPipeline.from_pretrained("./PixArt-XL-2-1024x1024", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 pipe.enable_attention_slicing()
 prompt = "A alpaca made of colorful building blocks, cyberpunk"

 ```python
 from diffusers import PixArtAlphaPipeline
 import torch
+pipe = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-1024-MS", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 pipe.enable_attention_slicing()
 prompt = "A alpaca made of colorful building blocks, cyberpunk"