23 lines
837 B
Python
23 lines
837 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"]
|
||
|
|
|
||
|
|
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")
|