Aryan 57ac673802 Refactor OmniGen (#10771) ...

* OmniGen model.py

* update OmniGenTransformerModel

* omnigen pipeline

* omnigen pipeline

* update omnigen_pipeline

* test case for omnigen

* update omnigenpipeline

* update docs

* update docs

* offload_transformer

* enable_transformer_block_cpu_offload

* update docs

* reformat

* reformat

* reformat

* update docs

* update docs

* make style

* make style

* Update docs/source/en/api/models/omnigen_transformer.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* update docs

* revert changes to examples/

* update OmniGen2DModel

* make style

* update test cases

* Update docs/source/en/api/pipelines/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* Update docs/source/en/using-diffusers/omnigen.md

Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>

* update docs

* typo

* Update src/diffusers/models/embeddings.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/models/attention.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/models/transformers/transformer_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/models/transformers/transformer_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/models/transformers/transformer_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update tests/pipelines/omnigen/test_pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update tests/pipelines/omnigen/test_pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <hlky@hlky.ac>

* consistent attention processor

* updata

* update

* check_inputs

* make style

* update testpipeline

* update testpipeline

* refactor omnigen

* more updates

* apply review suggestion

---------

Co-authored-by: shitao <2906698981@qq.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: hlky <hlky@hlky.ac>

2025-02-12 14:06:14 +05:30

..

allegro_transformer3d.md

Update variable names correctly in docs (#10435)

2025-01-06 08:56:43 -08:00

asymmetricautoencoderkl.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

aura_flow_transformer2d.md

[Docs] add AuraFlow docs (#8851)

2024-07-12 09:52:18 +02:00

autoencoder_dc.md

[chore] add contribution note for lawrence. (#10253)

2024-12-17 09:17:40 +05:30

autoencoder_kl_hunyuan_video.md

Community hosted weights for diffusers format HunyuanVideo weights (#10344)

2024-12-22 15:26:28 -10:00

autoencoder_oobleck.md

Stable Audio integration (#8716)

2024-07-30 15:29:06 +05:30

autoencoder_tiny.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

autoencoderkl_allegro.md

[core] Allegro T2V (#9736)

2024-10-29 13:14:36 +05:30

autoencoderkl_cogvideox.md

Add CogVideoX text-to-video generation model (#9082)

2024-08-06 21:23:57 -10:00

autoencoderkl_ltx_video.md

Rename LTX blocks and docs title (#10213)

2024-12-23 15:29:10 +05:30

autoencoderkl_mochi.md

[core] Mochi T2V (#9769)

2024-11-05 20:33:41 +05:30

autoencoderkl.md

Remove legacy single file model loading mixins (#8754)

2024-07-01 07:20:19 -10:00

cogvideox_transformer3d.md

Update variable names correctly in docs (#10435)

2025-01-06 08:56:43 -08:00

cogview3plus_transformer2d.md

Update variable names correctly in docs (#10435)

2025-01-06 08:56:43 -08:00

consisid_transformer3d.md

[core] ConsisID (#10140)

2025-01-19 13:10:08 +05:30

consistency_decoder_vae.md

[Docs] Fix typos (#7131)

2024-02-29 13:03:01 -08:00

controlnet_flux.md

[Flux] Support Union ControlNet (#9175)

2024-08-25 00:24:21 -10:00

controlnet_hunyuandit.md

[Tencent Hunyuan Team] Add Hunyuan-DiT ControlNet Inference (#8694)

2024-06-26 00:43:03 -10:00

controlnet_sd3.md

[Core] introduce controlnet module (#8768)

2024-11-06 22:08:55 -04:00

controlnet_sparsectrl.md

[core] AnimateDiff SparseCtrl (#8897)

2024-07-26 17:46:05 +05:30

controlnet_union.md

Add ControlNetUnion (#10131)

2024-12-11 07:07:50 -10:00

controlnet.md

[Core] introduce controlnet module (#8768)

2024-11-06 22:08:55 -04:00

dit_transformer2d.md

Update transformer2d.md title (#8375)

2024-06-03 17:01:21 -07:00

flux_transformer.md

Flux pipeline (#9043)

2024-08-01 11:30:52 -10:00

hunyuan_transformer2d.md

[HunyuanDiT] minor docs changes in hunyuandit (#8395)

2024-06-04 12:18:53 +04:00

hunyuan_video_transformer_3d.md

Community hosted weights for diffusers format HunyuanVideo weights (#10344)

2024-12-22 15:26:28 -10:00

latte_transformer3d.md

Latte: Latent Diffusion Transformer for Video Generation (#8404)

2024-07-11 15:06:22 +05:30

ltx_video_transformer3d.md

Rename LTX blocks and docs title (#10213)

2024-12-23 15:29:10 +05:30

lumina2_transformer2d.md

Add support for lumina2 (#10642)

2025-02-11 11:38:33 -10:00

lumina_nextdit2d.md

[Alpha-VLLM Team] Add Lumina-T2X to diffusers (#8652)

2024-07-07 17:12:09 -10:00

mochi_transformer3d.md

Update variable names correctly in docs (#10435)

2025-01-06 08:56:43 -08:00

omnigen_transformer.md

Refactor OmniGen (#10771)

2025-02-12 14:06:14 +05:30

overview.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

pixart_transformer2d.md

Errata - Trim trailing white space in the whole repo (#8575)

2024-06-24 18:39:15 +05:30

prior_transformer.md

Update transformer2d.md title (#8375)

2024-06-03 17:01:21 -07:00

sana_transformer2d.md

[Sana bug] bug fix for 2K model config (#10340)

2024-12-23 08:56:25 +05:30

sd3_transformer2d.md

Errata - Trim trailing white space in the whole repo (#8575)

2024-06-24 18:39:15 +05:30

stable_audio_transformer.md

Stable Audio integration (#8716)

2024-07-30 15:29:06 +05:30

stable_cascade_unet.md

[Docs] add stable cascade unet doc. (#9066)

2024-08-05 21:28:48 +05:30

transformer2d.md

[Chore] remove deprecation from transformer2d regarding the output class. (#8698)

2024-06-26 07:35:36 -10:00

transformer_temporal.md

Update transformer2d.md title (#8375)

2024-06-03 17:01:21 -07:00

unet2d-cond.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

unet2d.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

unet3d-cond.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

unet-motion.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

unet.md

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

uvit2d.md

[docs] UViT2D (#6643)

2024-01-25 09:37:28 -08:00

vq.md

move vqmodel to models.autoencoders. (#8292)

2024-05-29 06:30:35 +05:30