BAAI / Emu

Emu is a Large Multimodal Model (LMM) trained with a unified autoregressive objective, i.e. , predict-the-next-element, including both visual embeddings and textual tokens. Trained under this objective, Emu can serve as a generalist interface for diverse multimodal tasks, such as image captioning, image/video question answering, and text-to-image generation, together with new abilities like in-context text and image generation, and image blending.

Setup

Clone the github repository and install required packages:

git clone https://github.com/baaivision/Emu
cd Emu

pip install -r requirements.txt

Model Weights

We release the pretrained and instruction-tuned weights of Emu . Our weights are subject to LLaMA's license .

Model Usage

At present, we provide inference code for image captioning and visual question answering:

python emu_inference.py --instruct --ckpt-path $Instruct_CKPT_PATH

Acknowledgement

We thank the great work from LLaMA , BLIP-2 , Stable Diffusion , and FastChat .

Citation

If you find Emu useful for your your research and applications, please consider citing:

@article{Emu,
  title={Generative Pretraining in Multimodality},
  author={Sun, Quan and Yu, Qiying and Cui, Yufeng and Zhang, Fan and Zhang, Xiaosong and Wang, Yueze and Gao, Hongcheng and Liu, Jingjing and Huang, Tiejun and Wang, Xinlong},
  publisher={arXiv:2307.05222},
  year={2023},
}