EveryDream2trainer/utils/get_yamls.py

23 lines
837 B
Python
Raw Normal View History

2022-12-17 20:32:48 -07:00
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")