Cuda out of memory when trying to swap from only 2 images

neuralchen / SimSwap

An arbitrary face-swapping framework on images and videos with one single trained model!

Other

4.45k stars 881 forks source link

Cuda out of memory when trying to swap from only 2 images #393

Open MordechaiHadad opened 1 year ago

MordechaiHadad commented 1 year ago

full message: RuntimeError: CUDA out of memory. Tried to allocate 2.00 GiB (GPU 0; 6.00 GiB total capacity; 3.25 GiB already allocated; 0 bytes free; 4.51 GiB reserved in total by PyTorch)

What can I do about this?