exo
exo copied to clipboard
Add FLUX support with MLX backend
Related to #461
Add support for the FLUX model with the MLX backend.
-
New Model Class: Add
exo/inference/mlx/models/flux.pyto define theFluxModelandModelclasses for the FLUX model. -
Inference Engine Update: Modify
exo/inference/mlx/sharded_inference_engine.pyto include necessary imports for the FLUX model. -
Testing: Add
exo/inference/mlx/test_sharded_flux.pyto include tests for the FLUX model, ensuring proper integration and functionality. -
Documentation: Update
README.mdto include instructions for using the FLUX model with the MLX backend, along with usage examples.
Did you test this?
I don't think this works. Please re-open if that's not that case.