Img to Img taking way too much memory.
#2
by
Nolyzlel
- opened
Hello there,
using the model for normal Img generation works great.
However when trying to use Img to Img it takes a great deal of mem.
Error message reads: "OutOfMemoryError: CUDA out of memory. Tried to allocate 422.31 GiB (GPU 0; 14.58 GiB total capacity; 2.28 GiB already allocated; 11.12 GiB free; 2.32 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"
running on t4 with 16 gig mem.