Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,481 Bytes
3542be4 31877a7 5836c22 c458cb3 5836c22 3542be4 31877a7 3542be4 31877a7 63144e1 3542be4 31877a7 f1a09c7 5836c22 bc07ae3 5836c22 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import os
import requests
from tqdm import tqdm
import shutil
from PIL import Image, ImageOps
import numpy as np
import cv2
def dl_cn_model(model_dir):
folder = model_dir
file_name = 'diffusion_pytorch_model.safetensors'
url = "https://huggingface.co/2vXpSwA7/iroiro-lora/resolve/main/test_controlnet2/CN-anytest_v4-marged.safetensors"
file_path = os.path.join(folder, file_name)
if not os.path.exists(file_path):
response = requests.get(url, allow_redirects=True)
if response.status_code == 200:
with open(file_path, 'wb') as f:
f.write(response.content)
print(f'Downloaded {file_name}')
else:
print(f'Failed to download {file_name}')
else:
print(f'{file_name} already exists.')
def dl_cn_config(model_dir):
folder = model_dir
file_name = 'config.json'
file_path = os.path.join(folder, file_name)
if not os.path.exists(file_path):
config_path = os.path.join(os.getcwd(), file_name)
shutil.copy(config_path, file_path)
def dl_tagger_model(model_dir):
model_id = 'SmilingWolf/wd-swinv2-tagger-v3'
files = [
'config.json', 'model.onnx', 'selected_tags.csv', 'sw_jax_cv_config.json'
]
if not os.path.exists(model_dir):
os.makedirs(model_dir)
for file in files:
file_path = os.path.join(model_dir, file)
if not os.path.exists(file_path):
url = f'https://huggingface.co/{model_id}/resolve/main/{file}'
response = requests.get(url, allow_redirects=True)
if response.status_code == 200:
with open(file_path, 'wb') as f:
f.write(response.content)
print(f'Downloaded {file}')
else:
print(f'Failed to download {file}')
else:
print(f'{file} already exists.')
def dl_lora_model(model_dir):
file_name = 'lineart.safetensors'
file_path = os.path.join(model_dir, file_name)
if not os.path.exists(file_path):
url = "https://huggingface.co/tori29umai/lineart/resolve/main/sdxl_BWLine.safetensors"
response = requests.get(url, allow_redirects=True)
if response.status_code == 200:
with open(file_path, 'wb') as f:
f.write(response.content)
print(f'Downloaded {file_name}')
else:
print(f'Failed to download {file_name}')
else:
print(f'{file_name} already exists.')
|