ATH-MaaS/Ovis-U1-3B
Ovis-U1
<div align="center"> <img src=https://cdn-uploads.huggingface.co/production/uploads/637aebed7ce76c3b834cea37/3IK823BZ8w-mz_QfeYkDn.png width="30%"/> </div>
<p align="center"> <a href="https://arxiv.org/abs/2506.23044"><img src="https://img.shields.io/badge/arXivpaper-2506.23044-b31b1b.svg" alt="arxiv"></a> <!-- <a href="https://github.com/AIDC-AI/Ovis-U1/blob/main/docs/OvisU1Report.pdf"><img src="https://img.shields.io/badge/Paper-TechReport-b31b1b" alt="paper"></a> --> <a href="https://github.com/AIDC-AI/Ovis-U1"><img src="https://img.shields.io/badge/GitHub-AIDC--AI/Ovis--U1-blue?style=flat&logo=github" alt="code"></a> <a href="https://huggingface.co/spaces/AIDC-AI/Ovis-U1-3B"><img src="https://img.shields.io/badge/๐จHFSpaces-AIDC--AI/Ovis--U1--3B-lightblack" alt="demo"></a> <a href="https://huggingface.co/AIDC-AI/Ovis-U1-3B"><img src="https://img.shields.io/badge/๐ค_Model-AIDC--AI/Ovis--U1--3B-yellow" alt="model"></a> </p>
<p align="left"> Building on the foundation of the Ovis series, Ovis-U1 is a 3-billion-parameter unified model that seamlessly integrates <b>multimodal understanding</b>, <b>text-to-image generation</b>, and <b>image editing</b> within a single powerful framework. </p>
<p align="center"> <img src="https://cdn-uploads.huggingface.co/production/uploads/636f4c6b5d2050767e4a1491/EmEEGmot9JzaBfHP2uWld.jpeg" width="95%"> <br> <em>The overall architecture of Ovis-U1 (cf. Fig.2 in our report).</em> </p>
๐ฆ Installation
Ovis-U1 has been tested with Python 3.10, Torch 2.4.0, Transformers 4.51.3, and DeepSpeed 0.15.4. For a comprehensive list of package dependencies, please consult the requirements.txt file.
git clone git@github.com:AIDC-AI/Ovis-U1.git
conda create -n ovis-u1 python=3.10 -y
conda activate ovis-u1
cd Ovis-U1
pip install -r requirements.txt
pip install -e .
๐ ๏ธ Inference
For single image understanding, please run
python test_img_to_txt.pyFor multi-image understanding, please run
python test_multi_img_to_txt.pyFor text-to-image, please run
python test_txt_to_img.py \
--height 1024 \
--width 1024 \
--steps 50 \
--seed 42 \
--txt_cfg 5 For image editing, please run
python test_img_edit.py \
--steps 50 \
--img_cfg 1.5 \
--txt_cfg 6 ๐ Performance
OpenCompass Multi-modal Academic Benchmarks
GenEval
๐ denotes using the rewritten prompts
DPG-Bench
ImgEdit-Bench
GEdit-Bench-EN
๐ Citation
If you find Ovis-U1 useful, please cite our paper:
@article{wang2025ovisu1,
title={Ovis-U1 Technical Report},
author={Wang, Guo-Hua and Zhao, Shanshan and Zhang, Xinjie and Cao, Liangfu and Zhan, Pengxin and Duan, Lunhao and Lu, Shiyin and Fu, Minghao and Zhao, Jianshan and Li, Yang and Chen, Qing-Guo},
journal={arXiv preprint arXiv:2506.23044},
year={2025}
}๐ Acknowledgments
The code is built upon Ovis and FLUX.
๐ License
The project is released under Apache License 2.0 (http://www.apache.org/licenses/LICENSE-2.0, SPDX-License-identifier: Apache-2.0).
๐จ Disclaimer
We used compliance checking algorithms during the training process, to ensure the compliance of the trained model to the best of our ability. Due to complex data and the diversity of language model usage scenarios, we cannot guarantee that the model is completely free of copyright issues or improper content. If you believe anything infringes on your rights or generates improper content, please contact us, and we will promptly address the matter.
