trellis_mac: unblock — dinov3 lvd1689m granted; add HF_HUB_DISABLE_XET + OMP guards
dinov3-vitl16-pretrain-lvd1689m access granted (the natural-image variant TRELLIS.2 needs — note the approval email named the sat493m satellite variant, but lvd1689m came through too). Added HF_HUB_DISABLE_XET=1 (same xet download bug as flux) + KMP_DUPLICATE_LIB_OK/OMP_NUM_THREADS=1 guards (torch + Metal-kernel omp). First local TRELLIS.2 generation now running. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
parent
f0bac67edb
commit
764ae3faf5
@ -1,5 +1,6 @@
|
|||||||
import argparse
|
import argparse
|
||||||
import json
|
import json
|
||||||
|
import os
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@ -34,8 +35,18 @@ cmd = [
|
|||||||
if p.get("no_texture"):
|
if p.get("no_texture"):
|
||||||
cmd.append("--no-texture")
|
cmd.append("--no-texture")
|
||||||
|
|
||||||
|
env = os.environ.copy()
|
||||||
|
# xet chunked downloader fails on these BFL/Meta repos ("Unable to parse string
|
||||||
|
# as hex hash value") — force the reliable HTTP path (same fix as flux_local).
|
||||||
|
env["HF_HUB_DISABLE_XET"] = "1"
|
||||||
|
# guard the torch + compiled-kernel OpenMP collision like sf3d does.
|
||||||
|
env["KMP_DUPLICATE_LIB_OK"] = "TRUE"
|
||||||
|
env.setdefault("OMP_NUM_THREADS", "1")
|
||||||
|
env.setdefault("MKL_NUM_THREADS", "1")
|
||||||
|
|
||||||
print("+", " ".join(cmd), flush=True)
|
print("+", " ".join(cmd), flush=True)
|
||||||
res = subprocess.run(cmd, cwd=str(VENDOR), stdout=sys.stdout, stderr=subprocess.STDOUT)
|
print("(first run downloads TRELLIS.2-4B + dinov3 + RMBG-2.0 weights)", flush=True)
|
||||||
|
res = subprocess.run(cmd, cwd=str(VENDOR), stdout=sys.stdout, stderr=subprocess.STDOUT, env=env)
|
||||||
if res.returncode != 0:
|
if res.returncode != 0:
|
||||||
sys.exit(res.returncode)
|
sys.exit(res.returncode)
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user