import os, urllib.request, pathlib

root = pathlib.Path("models")
(root / "checkpoints").mkdir(parents=True, exist_ok=True)
(root / "loras").mkdir(parents=True, exist_ok=True)

def wget(url, dest):
    if not dest.exists():
        print("Downloading", url)
        urllib.request.urlretrieve(url, dest)

wget(
    "https://huggingface.co/boopyfloopy/gonzalomoXLFluxPony_v40UnityXLDMD/blob/main/gonzalomoXLFluxPony_v40UnityXLDMD.safetensors",
    root / "checkpoints" / "gonzalomoXLFluxPony_v40UnityXLDMD.safetensors"
)
wget(
    "https://huggingface.co/AiWise/epiCPhoto-XL-LoRA-Derp2/blob/main/epiCPhotoXL-LoRA-Derp2.safetensors",
    root / "loras" / "epiCPhotoXL-Derp2.safetensors"
)