Hi, Firstly, thanks for your amazing work. I have an RTX 2060m with 6Gb of VRAM which isn't enough to run the model. But is there a batch size parameter that I can tweak so that I can fit it in this GPU? There is an option of the resolution, but I would like to fit for the original resolution of 512. Let me know if this is possible. Thank you!
Hi, Firstly, thanks for your amazing work. I have an RTX 2060m with 6Gb of VRAM which isn't enough to run the model. But is there a batch size parameter that I can tweak so that I can fit it in this GPU? There is an option of the resolution, but I would like to fit for the original resolution of 512. Let me know if this is possible. Thank you!