HiDream-I1 is a new open-source image generative foundation model with 17B parameters that achieves state-of-the-art image generation quality within seconds.

Models

We offer both the full version and distilled models. For more information about the models, please refer to the link under Usage.

Name	Script	Inference Steps	HuggingFace repo
HiDream-I1-Full	inference.py	50	🤗 HiDream-I1-Full
HiDream-I1-Dev	inference.py	28	🤗 HiDream-I1-Dev
HiDream-I1-Fast	inference.py	16	🤗 HiDream-I1-Fast

Quick Start

Please make sure you have installed Flash Attention. We recommend CUDA versions 12.4 for the manual installation.

pip install -r requirements.txt

Then you can run the inference scripts to generate images:

python -m hi_diffusers "A cat holding a sign that says 'hello world'"

# or you can specify the model
python -m hi_diffusers "A cat holding a sign that says 'hello world'" -m fast

Note: The inference script will automatically download meta-llama/Meta-Llama-3.1-8B-Instruct model files. If you encounter network issues, you can download these files ahead of time and place them in the appropriate cache directory to avoid download failures during inference.

Gradio Dashboard

We also provide a Gradio demo for interactive image generation. You can run the demo with:

pip install gradio
python gradio_demo.py

License

The code in this repository and the HiDream-I1 models are licensed under MIT License.