Update convert.py

convert.py

@@ -28,9 +28,9 @@ def convert_single(model_id: str, filename: str, model_type: str, sample_size: i
 
     if model_type == "v1":
         config_url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
-    elif model_type == "v2.0":
+    elif model_type == ["v2.0", "v2.1"] and sample_size == 512:
         config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml"
-    elif model_type == "v2.1":
+    elif model_type in ["v2.0", "v2.1"]:
         config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
 
     config_file = BytesIO(requests.get(config_url).content)