ai-toolkit/extensions_built_in/diffusion_models/omnigen2/src
Jaret Burkett 611969ec1f Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -06:00
..
models Added support for finetuning OmniGen2. 2025-06-25 13:58:16 -06:00
ops/triton Added support for finetuning OmniGen2. 2025-06-25 13:58:16 -06:00
pipelines Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -06:00
schedulers Added support for finetuning OmniGen2. 2025-06-25 13:58:16 -06:00
utils Added support for finetuning OmniGen2. 2025-06-25 13:58:16 -06:00
__init__.py Added support for finetuning OmniGen2. 2025-06-25 13:58:16 -06:00