huggingface / huggingface/diffusers

[Modular] support group offloading in auto offloading

Open
#14,346 0 comments 0 reactions 2 assignees Claimed by @sayakpaul View on GitHub
Dominant language
Python
Stars
34.5k
Forks
7.3k
Avg merge
3d 3h
Merged PRs (30d)
91

Description

Follow-up to #14304.

`enable_auto_cpu_offload` recommends group offloading for a model that OOMs with everything else already offloaded ("it does not fit on the device by itself"). but currently we cannot use group offloading and auto offloading togehter, should add support once #14304 is in

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.