LibreAI / Image / OmniGen
// image · VectorSpaceLab/OmniGen

OmniGen

OmniGen: one unified model for text-to-image, editing, subject-driven and identity-preserving generation

★ 4.3k stars⑂ 364 forksMITupdated 2025-12
▦ visual — this project's README ships no embeddable screenshot,
so here's a placeholder. The guide below is complete regardless.
// project facts

Project facts

OmniGen is an open-source image project by VectorSpaceLab (VectorSpaceLab/OmniGen on GitHub). It has 4,343 GitHub stars and 364 forks, is released under the MIT license, and was last updated on GitHub on 4 December 2025.

GitHub stars
4,343
Forks
364
License
MIT
Last updated on GitHub
4 December 2025
Category
Image
Listing
Full install guide
// how it works

How it works

  1. Prepare a GPU environment

    Create the documented Conda environment (Python 3.10.x) — OmniGen inference expects a CUDA GPU.

  2. Clone and install

    Clone the repo and install it in editable mode after installing the pinned PyTorch build.

  3. Load the model

    Load Shitao/OmniGen-v1 weights from Hugging Face through the OmniGen pipeline (or Diffusers).

  4. Generate with references

    Write a prompt, pass input images with guidance scales, and save your first edited or subject-driven image.

// features

Why people love it

Unified task coverage

Text-to-image, editing, subject-driven, identity-preserving, and image-conditioned generation in one model.

No extra plugins

Tasks that usually need ControlNet or IP-Adapter are handled natively by the framework.

In-prompt image references

Reference images inline with <|image_1|>-style placeholders and combine multiple inputs.

Guidance controls

Separate guidance_scale and img_guidance_scale parameters balance text and image influence.

Diffusers and Gradio paths

Use the model through Hugging Face Diffusers or run the included Gradio demo (app.py).

Fine-tuning included

LoRA and full-parameter training scripts (train.py with accelerate) adapt the model to your data.

Advertisement
// run it

Run it yourself

  1. Clone the repository

    Clone the project and enter its directory.

    git clone https://github.com/VectorSpaceLab/OmniGen.git
    cd OmniGen
  2. Set up the environment

    Create the Conda env, install the pinned CUDA PyTorch build from the README, then install the package.

    conda create -n omnigen python=3.10.13
    conda activate omnigen
    pip install torch==2.3.1+cu118 torchvision --extra-index-url https://download.pytorch.org/whl/cu118
    pip install -e .
  3. Run inference

    Load the pretrained model and generate with an image reference, following the README's examples.

    from OmniGen import OmniGenPipeline
    pipe = OmniGenPipeline.from_pretrained("Shitao/OmniGen-v1")
    images = pipe(prompt="Edit <|image_1|>", input_images=["input.png"], guidance_scale=2.5, img_guidance_scale=1.6, seed=0)
// use cases

What you can build

Advertisement
// faq

Questions, answered

Who maintains OmniGen?

VectorSpaceLab, the research team behind the OmniGen paper (arXiv 2409.11340).

Is OmniGen free?

The code is MIT-licensed and the weights are downloadable from Hugging Face; check the model card for the weights' own terms.

Does it need a paid API key?

No — it runs locally on your GPU with downloaded weights.

How does it compare to Stable Diffusion plus ControlNet?

OmniGen folds editing, subject-driven, and conditioned generation into one model with in-prompt image references, instead of a base model plus separate plugin networks.

How do I get started?

Create the Python 3.10 Conda env, install the pinned PyTorch build, `pip install -e .`, then run the OmniGenPipeline example from the README.

// keep exploring

Similar tools

Image

Bagel

ByteDance's open-source unified multimodal model for understanding, generation and image editing.

★ 6.2k starsApache-2.0
View guide →
Image

omnigen-vault

Infinite text-free AI image generator with a searchable multilingual vault

★ 35 starsMIT
View guide →
Image

mmagic

MMagic — OpenMMLab's generative-AI toolbox: diffusion, text-to-image, restoration and a model zoo.

★ 7.5k starsApache-2.0
View guide →
Image

HunyuanImage-3.0

Tencent's HunyuanImage-3.0 — a powerful native multimodal model for image generation.

★ 3.3k starsNOASSERTION
View guide →
Image

pydantic-ai

How Python does AI. Agents, realtime voice, image generation, embeddings. Every model, every interface, typed end to end.

★ 20.4k starsMIT
View guide →
Image

LLM-Hub

Run LLMs plus image, video and music generation locally on your phone

★ 603 starsCustom
View guide →
// alternatives

Alternatives in Image

The most-starred other Image projects in the LibreAI catalogue — the realistic shortlist if OmniGen is not the right fit.

Get OmniGen

Free and open — grab the source, star the repo, and start creating.