EveryDream2trainer/utils/get_yamls.py

28 lines
1022 B
Python

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"]
# download https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.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]}")
def isWindows():
return sys.platform.startswith('win')
if __name__ == '__main__':
download_all()
print("SD1.x and SD2.x yamls downloaded")