Instructpix2pix training guide please

🧨 Diffusers
1

i m trying to train Instructpix2pix on my Colab free tier …but i m getting error everytime i m trying i have 1000 sample of dataset heres the error i m getting
OutOfMemoryError: CUDA out of memory. Tried to allocate 58.00 MiB (GPU 0; 14.75 GiB total capacity; 13.21 GiB already allocated; 44.81 MiB free; 13.35 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

heres the training manual i m following line by line:
InstructPix2pixTraining

can anyone help me out how i can train this model i m a newbie please do suggest me .

2

hey @nicky007 it very likely requires more gpu memory than a colab free tier has

3

i understand but are there any hacks or tricks i can do to make this work i tried some like decreasing batch size and dataset size.tried using xformers…i m using pytorch 2.0.1+cu117 do i need to downgrade the pytorch version .?