search

vvictoryuki / FreeDoM

[ICCV 2023] Official PyTorch implementation for the paper "FreeDoM: Training-Free Energy-Guided Conditional Diffusion Model"
260 stars 9 forks source link

What is the minimum VRAM required to run this? #1

Closed ninjasaid2k closed 1 year ago

ninjasaid2k commented 1 year ago

What is the minimum VRAM required to run this? 24 GB?

vvictoryuki commented 1 year ago

Thank you for your attention to our work! Considering that our method requires adding the process of solving the gradient in each sampling step, the GPU memory overhead will be higher than the training required methods. In our current experiments, the GPU memory cost for face-related experiments is 12G, for ImageNet-related experiments is 8G, for Stable Diffusion experiments is 17G, and for ControlNet experiments is 20G. There may be room to optimize the memory cost, but we currently have no plans to do so.