diffusers
Pass device to enable_model_cpu_offload in maybe_free_model_hooks
#6937

Merged

Pass device to enable_model_cpu_offload in maybe_free_model_hooks #6937

sayakpaul merged 1 commit into huggingface:main from Disty0:main

Pass device to enable_model_cpu_offload in maybe_free_model_hooks

0f077600

sayakpaul commented on 2024-02-11

sayakpaul requested a review from

patrickvonplaten 1 year ago

sayakpaul requested a review from

yiyixuxu 1 year ago

yiyixuxu approved these changes on 2024-02-12

yiyixuxu requested a review from

DN6 1 year ago

sayakpaul merged 9254d1f3 into main 1 year ago

Reviewers

yiyixuxu

sayakpaul

patrickvonplaten

DN6

Assignees

No one assigned

Labels

None yet

Milestone

No milestone