diffusers
80bfe77b - Remove JAX/Flax (#14169)

Commit
99 days ago
Remove JAX/Flax (#14169) * Remove JAX/Flax support Deletes all Flax models, schedulers, pipelines, and loading utilities, the from_flax loading path, flax/jax packaging extras, Flax CI filters, Flax training examples, and Flax content in translated docs. Hub compatibility is preserved: Flax class names in old model_index.json files are still stripped to their PyTorch equivalents, and *.msgpack files in repos are still ignored during download. * Drop Flax class-name remapping from model_index.json loading Flax-saved checkpoint metadata is no longer recognized: class names like FlaxStableDiffusionPipeline in model_index.json now fail to resolve instead of being remapped to their PyTorch equivalents. *.msgpack files in hub repos are still ignored during download. * Restore Flax training examples The example scripts are standalone and still work with diffusers<=0.39.x. A warning note is added to each Flax README section pointing out the version requirement now that the library itself no longer ships Flax. * update * Repoint scheduler default-config tests off the Flax fixture * Repoint example training tests off the Flax fixture
Author
DN6 DN6
Parents
Loading