Can I select the GPU type based on the base model in python script ? - Runpod