Commit dcbff4b

committed

Add NNXFluxTransformer2DModel and NNX FLUX transformer blocks

1 parent 5393d56 commit dcbff4bCopy full SHA for dcbff4b

1 file changed

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

Comments

(0)