Luo-Yihong
/

yoso_pixart1024

Model card Files Files and versions Community

Luo-Yihong commited on Mar 19, 2024

Commit

5e7020c

•

1 Parent(s): 16fe608

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ PixArt-XL-2-1024-MS](https://huggingface.co/PixArt-alpha/PixArt-XL-2-1024-MS
 ## usage
 ```python
 import torch
-from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel, DPMSolverMultistepScheduler
 transformer = Transformer2DModel.from_pretrained(
     "Luo-Yihong/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')

 ## usage
 ```python
 import torch
+from diffusers import PixArtAlphaPipeline, LCMScheduler, Transformer2DModel
 transformer = Transformer2DModel.from_pretrained(
     "Luo-Yihong/yoso_pixart1024", torch_dtype=torch.float16).to('cuda')