PR #2536 Enabling gradient checkpointing for VAE

Enabling gradient checkpointing for VAE #2536

williamberman merged 14 commits into huggingface:main from aandyw:gradient-checkpointing-vae

aandyw changed the title ~~Enabling gradient checkpointing for VAE~~ [WIP] Enabling gradient checkpointing for VAE 2 years ago

aandyw changed the title ~~[WIP] Enabling gradient checkpointing for VAE~~ Enabling gradient checkpointing for VAE 2 years ago

aandyw marked this pull request as ready for review 2 years ago

aandyw changed the title ~~Enabling gradient checkpointing for VAE~~ [WIP] Enabling gradient checkpointing for VAE 2 years ago

aandyw changed the title ~~[WIP] Enabling gradient checkpointing for VAE~~ Enabling gradient checkpointing for VAE 2 years ago

aandyw changed the title ~~Enabling gradient checkpointing for VAE~~ [WIP] Enabling gradient checkpointing for VAE 2 years ago

patrickvonplaten commented on 2023-03-07

patrickvonplaten requested a review from

williamberman 2 years ago

aandyw commented on 2023-03-08

aandyw removed review request from

williamberman 2 years ago

aandyw requested a review from

patrickvonplaten 2 years ago

patrickvonplaten commented on 2023-03-13

updated black format

b7471a0c

aandyw force pushed from 5056d873 to b7471a0c 2 years ago

update black format

fd876d69

make style format

a7f9da1d

updated line endings

d023b5e9

update code formatting

de4f5cd0

patrickvonplaten commented on 2023-03-14

Update examples/research_projects/onnxruntime/text_to_image/train_tex…

dcb23405

Update src/diffusers/models/vae.py

7ed45135

Update src/diffusers/models/vae.py

7ec6ccff

Merge branch 'huggingface:main' into gradient-checkpointing-vae

e79224f4

williamberman commented on 2023-03-16

williamberman requested a review from

williamberman 2 years ago

added vae gradient checkpointing test

a4343f08

Merge branch 'huggingface:main' into gradient-checkpointing-vae

3db930bf

Merge branch 'gradient-checkpointing-vae' of https://github.com/Pie31…

49dca752

make style

5bbbf1c8

williamberman approved these changes on 2023-03-17

Merge branch 'main' into gradient-checkpointing-vae

ffbec5fd

williamberman changed the title ~~[WIP] Enabling gradient checkpointing for VAE~~ Enabling gradient checkpointing for VAE 2 years ago

williamberman merged 116f70cb into main 2 years ago

Reviewers

williamberman

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

diffusers Enabling gradient checkpointing for VAE #2536 Merged

Enabling gradient checkpointing for VAE #2536

diffusers
Enabling gradient checkpointing for VAE
#2536

Merged