Hugging Face's logo

Luo-Yihong
/
yoso_pixart512

Text-to-Image
Diffusers
Safetensors
English
Model card Files Community
Luo-Yihong commited on
Commit
7febe82
·
verified ·
1 Parent(s): 5c0c829

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -14,7 +14,7 @@ PixArt-XL-2-512x512](https://huggingface.co/PixArt-alpha/PixArt-XL-2-512x512), e
14
  ## usage
15
  ```python
16
  import torch
17
- from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel, DPMSolverMultistepScheduler
18
 
19
  transformer = Transformer2DModel.from_pretrained(
20
  "Luo-Yihong/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')
 
14
  ## usage
15
  ```python
16
  import torch
17
+ from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel
18
 
19
  transformer = Transformer2DModel.from_pretrained(
20
  "Luo-Yihong/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')