Instructions to use AIPeanutman/OpenSubject with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use AIPeanutman/OpenSubject with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("AIPeanutman/OpenSubject", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
File size: 477 Bytes
1ebd499 c1af824 1ebd499 fba9c60 | 1 2 3 4 5 6 7 8 9 | ---
license: apache-2.0
---
# OmniGen2-OpenSubject
This repository provides an **open-source baseline** model, obtained by fine-tuning **OmniGen2** on the sampled 500k **OpenSubject** corpus and 100k internal T2I samples to maintain prompt-following ability. The model is optimized for **subject-driven image generation**, while maintaining general **prompt-following** ability.
For usage examples of **OpenSubject**, please refer to: https://github.com/LAW1223/OpenSubject.
|