cjwbw / pixart-sigma

Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

  • Public
  • 6K runs
  • GitHub
  • Paper
  • License

Run time and cost

This model costs approximately $0.064 to run on Replicate, or 15 runs per $1, but this varies depending on your inputs. It is also open source and you can run it on your own computer with Docker.

This model runs on Nvidia A40 (Large) GPU hardware. Predictions typically complete within 89 seconds. The predict time for this model varies significantly based on the inputs.

Readme

PixArt-Σ: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation

This repo contains PyTorch model definitions, pre-trained weights and inference/sampling code for our paper exploring Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation. You can find more visualizations on our project page.