import sys
import requests

_V2V_URL = ["v2-inference-v.yaml","https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"]
_V2_URL = ["v2-inference.yaml","https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference.yaml"]
_V1_URL = ["v1-inference.yaml","https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"]

def download_all():
    list = [_V2V_URL,_V2_URL,_V1_URL]
    for file in list:
        get_yaml(file)


def get_yaml(file):
    res = requests.request(method="GET", url=file[1])
    with open(file[0],"wb") as f:
        f.write(res.content)
    print(f" downloaded: {file[0]}")

if __name__ == '__main__':
    download_all()
    print("SD1.x and SD2.x yamls downloaded")