Update convert.py

convert.py

@@ -28,10 +28,11 @@ 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", "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 in ["v2.0", "v2.1"]:
-        config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
+    elif model_type == "v2":
+        if sample_size == 512:
+            config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml"
+        else:
+            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)