diffusers
SDXL flax
#4254
Merged
Overview
Commits
45
Changes
View On GitHub

SDXL flax #4254

patrickvonplaten merged 45 commits into main from sdxl-flax
pcuenca
mar-muel support transformer_layers_per block in flax UNet
cbdd6d6c
mar-muel add support for text_time additional embeddings to Flax UNet
4c78659d
mar-muel rename attention layers for VAE
3aa31641
mar-muel add shape asserts when renaming attention layers
bc267e4b
mar-muel transpose VAE attention layers
ea0e675e
mar-muel add pipeline flax SDXL code [WIP]
1cc6c37b
mar-muel continue add pipeline flax SDXL code [WIP]
9771feb3
mar-muel cleanup
1137263d
pcuenca Working on JIT support
f02d7958
pcuenca pcuenca marked this pull request as draft 2 years ago
pcuenca Merge remote-tracking branch 'origin/main' into sdxl-flax
e6eaee09
HuggingFaceDocBuilderDev
mar-muel
mar-muel commented on 2023-07-25
pcuenca Fixing embeddings (untested)
ff46fa20
mar-muel
mar-muel commented on 2023-07-25
mar-muel
mar-muel commented on 2023-07-25
pcuenca Remove spurious line
dfc3c81a
pcuenca Shard guidance_scale when jitting.
484e5163
pcuenca Decode images
70e1058b
pcuenca
pcuenca commented on 2023-07-25
pcuenca Merge remote-tracking branch 'origin/main' into sdxl-flax
2620ff3f
pcuenca pcuenca marked this pull request as ready for review 2 years ago
pcuenca pcuenca requested a review from yiyixuxu yiyixuxu 2 years ago
pcuenca pcuenca requested a review from sayakpaul sayakpaul 2 years ago
pcuenca pcuenca requested a review from williamberman williamberman 2 years ago
pcuenca pcuenca requested a review from patrickvonplaten patrickvonplaten 2 years ago
entrpn
pcuenca Fix sharding
8e638026
pcuenca style
4fd885a6
patrickvonplaten
patrickvonplaten commented on 2023-07-28
patrickvonplaten
patrickvonplaten commented on 2023-07-28
pcuenca Refiner UNet can be loaded.
27d26ca9
pcuenca
pcuenca commented on 2023-07-30
pcuenca Refiner / img2img pipeline
61d93f4e
patrickvonplaten
patrickvonplaten commented on 2023-08-02
patrickvonplaten
patrickvonplaten commented on 2023-08-02
pcuenca Allow latent outputs from base and latent inputs in refiner
636d69ff
pcuenca Adapt to FlaxCLIPTextModelOutput
3a00bfe5
pcuenca Merge branch 'sdxl-flax' of github.com:huggingface/diffusers into sdx…
a6442f78
pcuenca
pcuenca commented on 2023-08-05
pcuenca Update Flax XL pipeline to FlaxCLIPTextModelOutput
cba4327b
pcuenca Merge remote-tracking branch 'origin/main' into sdxl-flax
a34ca772
pcuenca make fix-copies
4411befb
pcuenca make style
5dc45d83
entrpn
entrpn commented on 2023-08-08
patil-suraj add euler scheduler
dd64ffac
entrpn
pcuenca Merge remote-tracking branch 'origin' into sdxl-flax
3a3850fd
pcuenca Fix import
66fd7121
pcuenca Fix copies, comment unused code.
4ecbc3f5
patrickvonplaten
pcuenca Fix SDXL Flax imports
0a5be230
pcuenca
pcuenca commented on 2023-09-21
patrickvonplaten Fix euler discrete begin
2bcf68de
patrickvonplaten improve init import
a9f34043
patrickvonplaten finish
cba9f043
patrickvonplaten put discrete euler in init
2c8e77fe
patrickvonplaten fix flax euler
461be999
patrickvonplaten Merge branch 'sdxl-flax' of https://github.com/huggingface/diffusers …
a975fc04
patrickvonplaten Fix more
d02a50bd
patrickvonplaten fix
53d595af
patrickvonplaten make style
3e1ffba8
patrickvonplaten
patrickvonplaten
patrickvonplaten commented on 2023-09-22
patrickvonplaten correct init
e01da93d
patrickvonplaten correct init
4346ed39
patrickvonplaten
patrickvonplaten commented on 2023-09-22
patrickvonplaten
patrickvonplaten commented on 2023-09-22
patrickvonplaten
patrickvonplaten commented on 2023-09-22
patrickvonplaten
patrickvonplaten approved these changes on 2023-09-22
pcuenca Temporarily remove FlaxStableDiffusionXLImg2ImgPipeline
da220454
pcuenca
pcuenca commented on 2023-09-22
patrickvonplaten correct pipelines
c503a36b
patrickvonplaten finish
b52c56d0
patrickvonplaten
patrickvonplaten patrickvonplaten merged 3651b14c into main 2 years ago
pcuenca
patrickvonplaten patrickvonplaten deleted the sdxl-flax branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum