h3 nsfw lora mirror job

worker/lora_mirror2.py

@@ -0,0 +1,43 @@
+"""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)