Yujun-Shi / DragDiffusion

[CVPR2024, Highlight] Official code for DragDiffusion

Home Page:https://yujun-shi.github.io/projects/dragdiffusion.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How much memory is needed?

zewei-Zhang opened this issue · comments

My GPU is 12g and still out of memory.

Hi, currently, it requires around 14~15 GB GPU memory.
I will continue to improve the memory efficiency.

I am happy to hear you are continuing to improve the memory efficiency.
When could we expect a version with improved memory efficiency?

@zewei-Zhang @MarkWijkhuizen Hi, I finally got time to look into the code.
Now if you're dragging an input real image, you should be able to do it under 12GB GPU memory with 512x512 images.

However, for dragging an 512x512 generated images, it still requires about 13GB memory.
I'll keep try to reduce the memory usage in the future.