PR #2285 `enable_model_cpu_offload`

`enable_model_cpu_offload` #2285

patrickvonplaten merged 21 commits into main from enable-model-offload

enable_model_offload PoC

059d5efb

pcuenca commented on 2023-02-08

pcuenca commented on 2023-02-08

patrickvonplaten commented on 2023-02-08

Rename final_offload_hook

a4769c97

Invoke the vae forward hook manually.

93564810

Completely remove decoder.

15db019e

pcuenca requested a review from

patrickvonplaten 3 years ago

Style

3f6a7b99

patrickvonplaten commented on 2023-02-08

patrickvonplaten approved these changes on 2023-02-08

apply_forward_hook decorator

900594e8

Rename method.

822808e7

Style

a829ba85

pcuenca changed the title ~~`enable_model_offload`, proof of concept~~ `enable_model_cpu_offload` 3 years ago

Copy enable_model_cpu_offload

2064e589

patrickvonplaten commented on 2023-02-16

Merge remote-tracking branch 'origin/main' into enable-model-offload

29502157

Fix copies.

5102106d

Remove comment.

6b2cc17d

Fix copies

52d42d98

Missing import

76e2e9e4

Fix doc-builder style.

05d5278b

Merge branch 'main' into enable-model-offload

7de9bfbb

Merge main and fix again.

a1a9bed8

Add docs

470f640c

Fix docs.

f5989451

Add a couple of tests.

3eb523e3

style

8f1560f9

pcuenca requested a review from

patil-suraj 3 years ago

pcuenca requested a review from

yiyixuxu 3 years ago

pcuenca requested a review from

williamberman 3 years ago

patrickvonplaten merged 2777264e into main 3 years ago

patrickvonplaten deleted the enable-model-offload branch 3 years ago

Reviewers

patrickvonplaten

keturn

patil-suraj

yiyixuxu

williamberman

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

diffusers `enable_model_cpu_offload` #2285 Merged

`enable_model_cpu_offload` #2285

diffusers
`enable_model_cpu_offload`
#2285

Merged