Hi, I am using google colab to run the given code which has only 12 GB RAM and the runtime is crashing is there any alternate to avoid this