PR #1423 Flax support for Stable Diffusion 2

Flax support for Stable Diffusion 2 #1423

pcuenca merged 16 commits into main from flax-sd-2

Flax: start adapting to Stable Diffusion 2

76bb311c

pcuenca marked this pull request as draft 3 years ago

More changes.

a87c3d98

camenduru requested changes on 2022-11-25

attention_head_dim can be a tuple.

99242770

Fix typos

b684d8c8

camenduru commented on 2022-11-26

pcuenca marked this pull request as ready for review 3 years ago

pcuenca requested a review from

patil-suraj 3 years ago

pcuenca requested a review from

patrickvonplaten 3 years ago

patrickvonplaten commented on 2022-11-28

Add simple SD 2 integration test.

7782f50b

Add simple UNet integration tests for Flax.

40f75504

Apply suggestions from code review

0cf9dbdf

Merge remote-tracking branch 'origin/main' into flax-sd-2

0d375e91

Typos and style

084d83a3

Tests: verify jax is available.

ff84eccb

Style

93d833d4

Make flake happy

eee35b83

Remove typo.

783d8cd2

Simple Flax SD 2 pipeline tests.

a32af1f9

Import order

383dacd3

pcuenca commented on 2022-11-28

Remove unused import.

d99411be

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

anton-l 3 years ago

patrickvonplaten commented on 2022-11-29

patrickvonplaten approved these changes on 2022-11-29

pcuenca merged 4d1e4e24 into main 3 years ago

pcuenca deleted the flax-sd-2 branch 3 years ago

Reviewers

patrickvonplaten

camenduru

patil-suraj

anton-l

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

diffusers Flax support for Stable Diffusion 2 #1423 Merged

Flax support for Stable Diffusion 2 #1423

diffusers
Flax support for Stable Diffusion 2
#1423

Merged