Thanks for the project! It's amazing what it can generate.  I use this evironment: RunPod Pytorch 2.1 A100 PCIe 120/20 A40 is not enough. OOM.