Running Eart2Studio FCN3 and GFS on RTX A 6000

I am running earth2studio on my RTX A 6000 GPU, I am using FCN3 model and GFS data but get the out of memory error even though my GPU is free and not occupied. Can anyone help me in running the inference