Flux inference implementation#146

Merged

entrpn merged 31 commits intomainfrom

Feb 12, 2025

Commits on Jan 14, 2025

Commits on Jan 15, 2025

add clip text encoder test
jfacevedo-google
committed

Commits on Jan 22, 2025

remove transformers inside maxdiffusion, add transformers dependency. Start creating generation code for flux.
jfacevedo-google
committed
add double block to flux
jfacevedo-google
committed
forward pass for single double block.
jfacevedo-google
committed

Commits on Jan 23, 2025

trying to use scan.
jfacevedo-google
committed

Commits on Jan 24, 2025

add single stream block
jfacevedo-google
committed

Commits on Jan 29, 2025

finish transformer
jfacevedo-google
committed

Commits on Jan 31, 2025

add missing conversions of pt to jax weights.
jfacevedo-google
committed

Commits on Feb 1, 2025

support both dev and schnell loading. Images still incorrect.
jfacevedo-google
committed

Commits on Feb 3, 2025

Commits on Feb 12, 2025