InvokeAI icon indicating copy to clipboard operation
InvokeAI copied to clipboard
verified publisher icon invoke-ai

Make force free GPU memory work in img2img

Open addianto opened this issue 3 years ago • 0 comments

This PR attempts to ensure --free_gpu_mem option work in img2img.py.

The solution is similar to the procedure in txt2img.py. I added condition check to see if --free_gpu_mem is enabled. If so, then move the loaded diffusion model into the CPU (the RAM, I presume?).

addianto avatar Dec 07 '22 11:12 addianto