Downloads · 30 days
15
1% of all-time downloads
pixart-900m
Owen777/pixart-900m
pixart-900m is a text-to-image model from Owen777. Use it when you need an image from a text prompt. It is set up for diffusers. The card lists the license as apache-2.0.
All-time downloads
1.1K
Public
Parameters
908M
3.6 GB on disk
Likes
7
Public
Repo makeup
Click a slice to open those files.
.safetensors3.6 GB · 100%
At a glance
- Task
- Text-to-Image
- Library
- diffusers
- License
- apache-2.0
- Access
- Public
- Created
- Dec 9, 2024
- Updated
- Dec 9, 2024
- SHA
- 31a677bc
Try a prompt
Model card
From the Hugging Face model README
<Gallery />
PixArt-900M-DiT v0.1
This is an experimental 900M DiT model that has been fully fine-tuned on a large-scale, high-quality dataset of images aligned with human preferences. It uses the same VAE and text encoder as PixArt-Sigma, so you only need to replace the transformer.
How to use it?
weight_dtype = torch.bfloat16
device = torch.device("cuda")
MODEL_ID = "PixArt-alpha/pixart_sigma_sdxlvae_T5_diffusers"
transformer = Transformer2DModel.from_pretrained("Owen777/pixart-900m", torch_dtype=weight_dtype)
pipe = PixArtSigmaPipeline.from_pretrained(MODEL_ID, transformer=transformer, torch_dtype=weight_dtype)
pipe.vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=weight_dtype).cuda()
pipe.to(device)
image = pipe("a cute cat with pure white background.",guidance_scale=3.5,num_inference_steps=30,height=1024,width=102)[0]
image.save("result.jpeg")
Credits
Original PixArt Sigma model by PIXART-Σ (https://pixart-alpha.github.io/PixArt-sigma-project/)
This model is created by Tian Ye and Peijie Dong from ROAS/DSA, HKUST-GZ.
- Task
- Text-to-Image
- Library
- diffusers
- License
- apache-2.0
- Created
- Dec 9, 2024
- Updated
- Dec 9, 2024