DragDiffusion icon indicating copy to clipboard operation
DragDiffusion copied to clipboard

Published 20 hours ago • verified publisher icon Yujun-Shi

How much memory is needed?

Open zewei-Zhang opened this issue 1 year ago • 3 comments

My GPU is 12g and still out of memory.

zewei-Zhang avatar Jul 17 '23 03:07 zewei-Zhang

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

Yujun-Shi avatar Jul 17 '23 04:07 Yujun-Shi

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

MarkWijkhuizen avatar Jul 18 '23 11:07 MarkWijkhuizen

@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.

Yujun-Shi avatar Dec 09 '23 11:12 Yujun-Shi