modelbeast/scripts/install_trellis2_mlx.sh
2026-07-20 17:04:20 +10:00

21 lines
1.0 KiB
Bash
Executable File

#!/bin/bash
# Vendor the MRP MLX fork of TRELLIS.2 (monster/trellis-2-mrp-mlx) for the
# trellis2_mlx operator. Idempotent. Existing per-box clones also work
# (run.py resolves m3 staging / m1 ~/trellis2-mlx automatically) — this
# script is for fresh boxes.
# Weights: TRELLIS.2-4B + gated facebook/dinov3 + briaai/RMBG-2.0 — rsync
# from m3ultra's HF cache (no HF login):
# for M in models--facebook--dinov3-vitl16-pretrain-lvd1689m \
# models--microsoft--TRELLIS.2-4B models--briaai--RMBG-2.0; do
# rsync -a m3ultra@100.89.131.57:.cache/huggingface/hub/$M ~/.cache/huggingface/hub/
# done
set -euo pipefail
ROOT="$(cd "$(dirname "$0")/.." && pwd)"
DEST="$ROOT/vendor/trellis2-mlx"
REPO="ssh://git@100.71.119.27:222/monster/trellis-2-mrp-mlx.git"
[ -d "$DEST/.git" ] && echo "already cloned: $DEST" || git clone "$REPO" "$DEST"
cd "$DEST"
PB="${PYTHON_BIN:-$(/opt/homebrew/bin/uv python find 3.11)}"
PYTHON_BIN="$PB" bash scripts/setup_macos.sh
.venv/bin/python -c "import mlx_backend" && echo "trellis2-mlx OK: $DEST"