Upload README.md

README.md

@@ -0,0 +1,58 @@
+---
+frameworks:
+- Pytorch
+license: apache-2.0
+tasks:
+- any-to-any
+---
+
+## What is the Nexus-Gen
+Nexus-Gen is a unified model that synergizes the language reasoning capabilities of LLMs with the image synthesis power of diffusion models. To align the embedding space of the LLM and diffusion model, we conduct a dual-phase alignment training process. (1) The autoregressive LLM learns to predict image embeddings conditioned on multimodal inputs, while (2) the vision decoder is trained to reconstruct high-fidelity images from these embeddings. During training the LLM, we identified a critical discrepancy between the autoregressive paradigm's training and inference phases, where error accumulation in continuous embedding space severely degrades generation quality. To avoid this issue, we introduce a prefilled autoregression strategy that prefills input sequence with position-embedded special tokens instead of continuous embeddings. Through dual-phase training, Nexus-Gen has developed the integrated capability to comprehensively address the image understanding, generation and editing tasks as follows.
+
+More information please refer to our repo: https://github.com/modelscope/Nexus-Gen.git
+
+![cover](assets/illustrations/gen_edit.jpg)
+![architecture](assets/illustrations/architecture.png)
+
+## Getting Started
+### Installation
+1. Install [DiffSynth-Studio](https://github.com/modelscope/DiffSynth-Studio.git) from source:
+```shell
+git clone https://github.com/modelscope/DiffSynth-Studio.git
+cd DiffSynth-Studio
+pip install -e .
+```
+2. Install requirements
+```
+pip install -r requirements.txt
+```
+3. Install [ms-swift](https://github.com/modelscope/ms-swift.git) if you want to perform finetuning on Nexus-Gen.
+```
+pip install ms-swift -U
+```
+### Prepare models
+```shell
+python download_models.py
+```
+### Image Understanding
+```shell
+python image_understanding.py
+```
+
+### Image Generation
+image generation with detailed prompt.
+```shell
+python image_generation.py
+```
+Polish prompt and generate images with Nexus-Gen.
+```shell
+image_generation_with_selfpolish.py
+```
+
+### Image Editing
+```shell
+python image_editing.py
+```
+
+### Training Codes
+Nexus-Gen is trained base on [ms-swift](https://github.com/modelscope/ms-swift.git) and [DiffSynth-Studio](https://github.com/modelscope/DiffSynth-Studio.git). You can find the training scripts in `train/scripts/train_decoder.sh` and `train_llm.sh`.