Error when running + Where do I get the images?

rbbrdckybk / ai-art-generator

For automating the creation of large batches of AI-generated artwork locally.

Other

634 stars 128 forks source link

Error when running + Where do I get the images? #16

Closed vrotmelk closed 2 years ago

vrotmelk commented 2 years ago

RuntimeError: CUDA out of memory. Tried to allocate 72.00 MiB (GPU 0; 4.00 GiB total capacity; 3.35 GiB already allocated; 0 bytes free; 3.41 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

Also the images arent in the output folder.

rbbrdckybk commented 2 years ago

Looks like you have 4GB of VRAM - you'll need more than that. 8GB is a realistic minimum to create small images.

vrotmelk commented 2 years ago

So I wont be able to do it?

rbbrdckybk commented 2 years ago

I've added support for Stable Diffusion in the last few days, if you use that and set !SD_LOW_MEMORY = yes in your prompt file it should work at lower resolutions. (the VQGAN and CLIP-guided diffusion processes will definitely not work with 4GB VRAM).

You'll need to update your installation using the latest instructions starting at step 13.