exo icon indicating copy to clipboard operation
exo copied to clipboard

Add FLUX support with MLX backend

Open vishwamartur opened this issue 1 year ago • 1 comments

Related to #461

Add support for the FLUX model with the MLX backend.

  • New Model Class: Add exo/inference/mlx/models/flux.py to define the FluxModel and Model classes for the FLUX model.
  • Inference Engine Update: Modify exo/inference/mlx/sharded_inference_engine.py to include necessary imports for the FLUX model.
  • Testing: Add exo/inference/mlx/test_sharded_flux.py to include tests for the FLUX model, ensuring proper integration and functionality.
  • Documentation: Update README.md to include instructions for using the FLUX model with the MLX backend, along with usage examples.

vishwamartur avatar Nov 25 '24 14:11 vishwamartur

Did you test this?

AlexCheema avatar Nov 25 '24 17:11 AlexCheema

I don't think this works. Please re-open if that's not that case.

AlexCheema avatar Nov 28 '24 07:11 AlexCheema