diffusers

mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Pedro Cuenca 4d1e4e24e5 Flax support for Stable Diffusion 2 (#1423 )

* Flax: start adapting to Stable Diffusion 2

* More changes.

* attention_head_dim can be a tuple.

* Fix typos

* Add simple SD 2 integration test.

Slice values taken from my Ampere GPU.

* Add simple UNet integration tests for Flax.

Note that the expected values are taken from the PyTorch results. This
ensures the Flax and PyTorch versions are not too far off.

* Apply suggestions from code review

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* Typos and style

* Tests: verify jax is available.

* Style

* Make flake happy

* Remove typo.

* Simple Flax SD 2 pipeline tests.

* Import order

* Remove unused import.

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: @camenduru

2022-11-29 12:33:21 +01:00

fixtures/custom_pipeline

[Custom pipeline] Easier loading of local pipelines (#1327 )

2022-11-17 16:00:26 +01:00

models

Flax support for Stable Diffusion 2 (#1423 )

2022-11-29 12:33:21 +01:00

pipelines

Flax support for Stable Diffusion 2 (#1423 )

2022-11-29 12:33:21 +01:00

repo_utils

Move accelerate to a soft-dependency (#1134 )