patrickvonplaten commited on
Commit
ee4e9f1
1 Parent(s): 1f71274

Update convert.py

Browse files
Files changed (1) hide show
  1. convert.py +2 -2
convert.py CHANGED
@@ -28,9 +28,9 @@ def convert_single(model_id: str, filename: str, model_type: str, sample_size: i
28
 
29
  if model_type == "v1":
30
  config_url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
31
- elif model_type == "v2.0":
32
  config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml"
33
- elif model_type == "v2.1":
34
  config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
35
 
36
  config_file = BytesIO(requests.get(config_url).content)
 
28
 
29
  if model_type == "v1":
30
  config_url = "https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
31
+ elif model_type == ["v2.0", "v2.1"] and sample_size == 512:
32
  config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml"
33
+ elif model_type in ["v2.0", "v2.1"]:
34
  config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
35
 
36
  config_file = BytesIO(requests.get(config_url).content)