Spaces:
Sleeping
Sleeping
File size: 2,434 Bytes
6b16ae8 1727c65 6b16ae8 1727c65 6b16ae8 1727c65 6b16ae8 abcf9af 6b16ae8 1727c65 |
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 |
import os
import subprocess
from huggingface_hub import hf_hub_download
model_dir_stablediffusion = "models/Stable-diffusion/"
os.makedirs(model_dir_stablediffusion, exist_ok=True)
model_dir_controlnet = "models/ControlNet/"
os.makedirs(model_dir_controlnet, exist_ok=True)
model_dir_ESRGAN = "models/ESRGAN"
os.makedirs(model_dir_ESRGAN, exist_ok=True)
model_dir_VAEapprox = "models/VAE-approx"
os.makedirs(model_dir_VAEapprox, exist_ok=True)
model_dir_lora = "models/Lora"
os.makedirs(model_dir_lora, exist_ok=True)
model_dir_karlo = "models/karlo"
os.makedirs(model_dir_karlo, exist_ok=True)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="juggernaut_reborn.safetensors", local_dir=model_dir_stablediffusion)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="control_v11f1e_sd15_tile.pth", local_dir=model_dir_controlnet)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="4x-UltraSharp.pth", local_dir=model_dir_ESRGAN)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="model.pt", local_dir=model_dir_VAEapprox)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="more_details.safetensors", local_dir=model_dir_lora)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="SDXLrender_v2.0.safetensors", local_dir=model_dir_lora)
model_path = hf_hub_download(repo_id="PietroC01/ImgEnhancerModels", filename="ViT-L-14_stats.th", local_dir=model_dir_karlo)
# Clona le estensioni se non esistono già
extensions_dir = "extensions"
os.makedirs(extensions_dir, exist_ok=True)
# Clona multidiffusion-upscaler
multidiffusion_dir = os.path.join(extensions_dir, "multidiffusion-upscaler-for-automatic1111")
if not os.path.exists(multidiffusion_dir):
subprocess.run(["git", "clone", "https://github.com/pkuliyi2015/multidiffusion-upscaler-for-automatic1111.git", multidiffusion_dir])
# Clona controlnet
controlnet_dir = os.path.join(extensions_dir, "sd-webui-controlnet")
if not os.path.exists(controlnet_dir):
subprocess.run(["git", "clone", "https://github.com/Mikubill/sd-webui-controlnet.git", controlnet_dir])
# Hugging Face Spaces usa la variabile d'ambiente PORT
port = os.getenv("PORT", "7860")
# Aggiungi il parametro --skip-torch-cuda-test per forzare l'uso della CPU
subprocess.run(["python3", "launch.py", "--skip-torch-cuda-test", "--port", port]) |