PR #2039 Dreambooth: reduce VRAM usage

Dreambooth: use `optimizer.zero_grad(set_to_none=True)` to reduce VRAM usage

gleb-akhmerov committed 2 years ago

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

gleb-akhmerov committed 2 years ago

Allow the user to control `optimizer.zero_grad(set_to_none=True)` with --set_grads_to_none

gleb-akhmerov committed 2 years ago

Update Dreambooth readme

gleb-akhmerov committed 2 years ago

Fix link in readme

gleb-akhmerov committed 2 years ago

Fix header size in readme

gleb-akhmerov committed 2 years ago

diffusers Dreambooth: reduce VRAM usage #2039 Merged