PR #2039 Dreambooth: reduce VRAM usage

Dreambooth: reduce VRAM usage #2039

patil-suraj merged 6 commits into huggingface:main from gleb-akhmerov:patch-2

Dreambooth: use `optimizer.zero_grad(set_to_none=True)` to reduce VRA…

31c0a2b2

gleb-akhmerov changed the title ~~Dreambooth: use `optimizer.zero_grad(set_to_none=True)` to reduce VRAM usage~~ Dreambooth: reduce VRAM usage 2 years ago

patrickvonplaten requested a review from

patil-suraj 2 years ago

patil-suraj commented on 2023-01-23

Merge remote-tracking branch 'origin/main' into patch-2

6bc0ce9e

Allow the user to control `optimizer.zero_grad(set_to_none=True)` wit…

3f63b16a

Update Dreambooth readme

e9872334

Fix link in readme

63061927

Fix header size in readme

496150b7

patil-suraj approved these changes on 2023-01-23

patil-suraj merged a66f2bae into main 2 years ago

Reviewers

patil-suraj

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

diffusers Dreambooth: reduce VRAM usage #2039 Merged

Dreambooth: reduce VRAM usage #2039

diffusers
Dreambooth: reduce VRAM usage
#2039

Merged