HiDream-I1 HiDream-I1 is a new open-source image generative foundation model with 17B parameters that achieves state-of-the-art image gener…
HiDream-I1 HiDream-I1 is a new open-source image generative foundation model with 17B parameters that achieves state-of-the-art image gener…
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.
For more features and to experience the full capabilities of our product, please visit https://vivago.ai/.
diffusers library. Check out the docs here.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-FastPlease make sure you have installed Flash Attention. We recommend CUDA versions 12.4 for the manual installation.
pip install -r requirements.txt
pip install -U flash-attn --no-build-isolation
Then you can run the inference scripts to generate images:
# For full model inference
python ./inference.py --model_type full
# For distilled dev model inference
python ./inference.py --model_type dev
# For distilled fast model inference
python ./inference.py --model_type fast
[!NOTE] The inference script will try to automatically download
meta-llama/Llama-3.1-8B-Instructmodel files. You need to agree to the license of the Llama model on your HuggingFace account and login usinghuggingface-cli loginin order to use the automatic downloader.
We also provide a Gradio demo for interactive image generation. You can run the demo with:
python gradio_demo.py
We recommend install Diffusers from source for better compatibility.
pip install git+https://github.com/huggingface/diffusers.git
Then you can inference HiDream-I1 with the following command:
…
The code in this repository and the HiDream-I1 models are licensed under MIT License.
@article{hidreami1technicalreport,
title={HiDream-I1: A High-Efficient Image Generative Foundation Model with Sparse Diffusion Transformer},
author={Cai, Qi and Chen, Jingwen and Chen, Yang and Li, Yehao and Long, Fuchen and Pan, Yingwei and Qiu, Zhaofan and Zhang, Yiheng and Gao, Fengbin and Xu, Peihan and others},
journal={arXiv preprint arXiv:2505.22705},
year={2025}
}
No open issues yet, or sync has not completed.