Hugging Face's logo

diffusers
/
controlnet-zoe-depth-sdxl-1.0

Text-to-Image
Diffusers
Safetensors
stable-diffusion-xl
stable-diffusion-xl-diffusers
controlnet
Model card Files
xet
 Community
6

Fix cpu offloading

#5
by tolgacangoz - opened
base: refs/heads/main
←
from: refs/pr/5
Discussion Files changed
+4
-4
Fix cpu offloadf1b17170
tolgacangoz
•
edited Aug 16, 2024

@sayakpaul @a-r-r-o-w

a-r-r-o-w

Thanks a lot! These look correct to me and will save the additional moves from cpu -> cuda, and then back to cpu due to offloading. Will wait for @sayakpaul to review once as well
tolgacangoz changed pull request title from Fix cpu offload to Fix cpu offloading
tolgacangoz

Similar to this:
https://github.com/huggingface/diffusers/pull/9207/files#diff-b31ccb92f4684fa9a9880dfe9001124738f7b4f90f40a5803677666dfbfbdec0
a-r-r-o-w changed pull request status to merged
tolgacangoz

Thanks for merging!

Sign up or log in to comment