File size: 1,677 Bytes
3b54ee9 | 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 | """One-shot: mirror the H3 nsfw LoRAs from civitai to HF (token-gated files)."""
import os
import time
import urllib.request
from huggingface_hub import HfApi
REPO = "nikdevs/minimax-h3-loras"
TOKEN = os.environ["CIVITAI_TOKEN"]
FILES = {
# HMPussy v0.5 — both files needed together
"hmpussy/vagassist_e40.safetensors":
"https://civitai.com/api/download/models/3215304?type=Model&format=SafeTensor&fileId=3097094",
"hmpussy/hmpussy_v6_epoch30.safetensors":
"https://civitai.com/api/download/models/3215304?type=Model&format=SafeTensor&fileId=3097100",
# H3 Riding POV i2v v1.0 (the H3 one, NOT the LTXV versions)
"riding/riding_pose_H3_i2v_v1.0.safetensors":
"https://civitai.com/api/download/models/3203205?fileId=3084588",
}
api = HfApi()
api.create_repo(REPO, exist_ok=True, private=False)
existing = {s.rfilename for s in api.repo_info(REPO).siblings}
for name, url in FILES.items():
if name in existing:
print(f"[skip] {name}", flush=True)
continue
t0 = time.time()
print(f"[dl] {name}", flush=True)
req = urllib.request.Request(url + f"&token={TOKEN}", headers={"User-Agent": "Mozilla/5.0"})
local = "/tmp/" + name.split("/")[-1]
with urllib.request.urlopen(req, timeout=600) as r, open(local, "wb") as f:
while chunk := r.read(1 << 22):
f.write(chunk)
print(f"[dl] {os.path.getsize(local)//1_000_000}MB in {time.time()-t0:.0f}s, uploading", flush=True)
api.upload_file(path_or_fileobj=local, path_in_repo=name, repo_id=REPO)
os.remove(local)
print(f"[up] {name} done", flush=True)
print("H3 LORA MIRROR DONE", flush=True)
time.sleep(10 ** 6)
|