Commit 63f7d0b

committed

Add NNXFluxTransformer2DModel and NNX FLUX transformer blocks

1 parent 38a7917 commit 63f7d0bCopy full SHA for 63f7d0b

1 file changed

src/maxdiffusion/models/flux/transformers
- transformer_flux_flax.py

Comments

(0)