nerdyrodent / VQGAN-CLIP

Just playing with getting VQGAN+CLIP running locally, rather than having to use colab.
Other
2.59k stars 428 forks source link

CUDA out of memory #150

Closed Tobe2d closed 2 years ago

Tobe2d commented 2 years ago

I am getting this error: RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 11.00 GiB total capacity; 10.11 GiB already allocated; 0 bytes free; 10.16 GiB reserved in total by PyTorch)

When I try to run this: python generate.py -p "The inside of a sphere" -zvid -i 4500 -zse 20 -vl 10 -zsc 0.97 -opt Adagrad -lr 0.15 -se 4500

How can I Fix this? change and make it work? I am running on 2080 Ti 11 GB VRAM

nerdyrodent commented 2 years ago

See https://github.com/nerdyrodent/VQGAN-CLIP#runtimeerror-cuda-out-of-memory