diffusers
let's go Flux2 🚀
#12711
Merged

let's go Flux2 🚀 #12711

sayakpaul merged 76 commits into main from full-flux2
sayakpaul
yiyixuxu add vae
fce87b36
dg845 Initial commit for Flux 2 Transformer implementation
a1f2ba1a
yiyixuxu add pipeline part
e470643e
sayakpaul small edits to the pipeline and conversion
7456a498
yiyixuxu Merge pull request #3 from huggingface/flux2-small-edits
f3405998
yiyixuxu update conversion script
b10e6296
yiyixuxu fix
73823584
yiyixuxu Merge branch 'flux2' of github.com:huggingface/diffusers-new-model-ad…
78ea69d6
yiyixuxu up up
524b1238
yiyixuxu finish pipeline
8a48adcd
dg845 Remove Flux IP Adapter logic for now
429d2cf5
dg845 Remove deprecated 3D id logic
2d7bad73
dg845 Remove ControlNet logic for now
c67f5820
dg845 Add link to ViT-22B paper as reference for parallel transformer block…
7acd7dad
update pipeline
66040866
dg845 Don't use biases for input projs and output AdaNorm
a200780f
up
f7148a02
dg845 Remove bias for double stream block text QKV projections
54c60803
dg845 Add script to convert Flux 2 transformer to diffusers
dba4c1fe
dg845 make style and make quality
cceffc4b
sayakpaul fix a few things.
e93a7461
sayakpaul allow sft files to go.
cfdd0057
yiyixuxu Merge pull request #4 from huggingface/flux2-model-cpu
edacb522
fix image processor
14839807
fix batch
62231ef9
style a bit
68db1782
dg845 Merge pull request #5 from huggingface/small-flux2-transformer-fix
89e42d94
dg845 Fix some bugs in Flux 2 transformer implementation
f6c82a31
dg845 Fix dummy input preparation and fix some test bugs
4082c430
sayakpaul fix dtype casting in timestep guidance module.
2b5b2e33
dg845 Merge pull request #6 from huggingface/fix-dtype
247770fd
sayakpaul resolve conflicts.,
3d022b8b
sayakpaul remove ip adapter stuff.
ffb00619
dg845 Fix Flux 2 transformer consistency test
6820d6ca
dg845 Fix bug in Flux2TransformerBlock (double stream block)
f6059b75
sayakpaul merge
5a108f0c
dg845 Get remaining Flux 2 transformer tests passing
05fcddbf
dg845 make style; make quality; make fix-copies
ac4f61a1
sayakpaul remove stuff.
20f9b830
sayakpaul Merge branch 'flux2-transformer' into full-flux2
e8d71ab4
sayakpaul fix type annotaton.
546e60da
sayakpaul remove unneeded stuff from tests
eeb52c28
sayakpaul tests
771e17bc
sayakpaul up
28da6799
sayakpaul up
14f98663
DN6 add sf support
6d169759
dg845 Remove unused IP Adapter and ControlNet logic from transformer (#9)
be6604bb
sayakpaul copied from
ec0a2add
sayakpaul Apply suggestions from code review
980bcc86
sayakpaul up
e1d46ce4
sayakpaul up
de993441
sayakpaul up
739c31b8
sayakpaul up
ded81a9c
sayakpaul up
c0086ea1
dg845 Refactor Flux2Attention into separate classes for double stream and s…
4f1f67a8
dg845 Add _supports_qkv_fusion to AttentionModuleMixin to allow subclasses …
b8e37600
dg845 Have Flux2ParallelSelfAttention inherit from AttentionModuleMixin wit…
24159acc
dg845 Log debug message when calling fuse_projections on a AttentionModuleM…
cb21400c
dg845 Address review comments
ba52a591
dg845 Merge pull request #11 from huggingface/fix-fused-projections
cc74755d
sayakpaul Update src/diffusers/pipelines/flux2/pipeline_flux2.py
697a43c9
sayakpaul up
aba48dde
dg845 Remove maybe_allow_in_graph decorators for Flux 2 transformer blocks …
963ed57b
sayakpaul up
997dfc24
sayakpaul Merge branch 'main' into full-flux2
842a8496
sayakpaul support ostris loras. (#13)
8d8bb3da
sayakpaul up
454cef8d
sayakpaul Merge branch 'main' into full-flux2
6bf955aa
update schdule
fc1bd891
sayakpaul up
29b02b80
yiyixuxu Merge pull request #15 from huggingface/final-update
24ff10e3
DN6 Merge pull request #14 from huggingface/transformer-single-file-support
4ef0285b
sayakpaul up (#17)
823f4c3f
sayakpaul add training scripts (#16)
5419877c
sayakpaul up
0a5a4773
sayakpaul sayakpaul requested a review from DN6 DN6 219 days ago
sayakpaul sayakpaul requested a review from yiyixuxu yiyixuxu 219 days ago
sayakpaul sayakpaul requested a review from dg845 dg845 219 days ago
sayakpaul Merge branch 'main' into full-flux2
cb9f1246
yiyixuxu
yiyixuxu approved these changes on 2025-11-25
HuggingFaceDocBuilderDev
DN6
DN6 approved these changes on 2025-11-25
sayakpaul
sayakpaul
sayakpaul sayakpaul merged 5ffb73d4 into main 219 days ago
sayakpaul sayakpaul deleted the full-flux2 branch 219 days ago
rentainhe
okaris

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone