FluxPipeline error while loading Flux.1 dev

🧨 Diffusers
7

Hi,

Could you remove Jax from your environment? pip uninstall jax. Alternatively, you could try upgrading jax, pip install --upgrade jax.

The code snippet only requires PyTorch and Diffusers to be installed.