scripts/run_trellis_detached.sh

#!/bin/bash
# Generate a textured mesh from the SAM-cropped LEGO brick with TRELLIS.2.
#
# Detached on purpose: this outlives the shell that launches it, so a closed
# session does not kill a multi-minute GPU job. Launch with:
#
#     setsid bash scripts/run_trellis_detached.sh > outputs/trellis_run.log 2>&1 < /dev/null &
#
# Watch it with:  tail -f outputs/trellis_run.log
# Result:         outputs/trellis_brick.glb  (+ .json sidecar with the stats)
set -uo pipefail

REPO=/srv/data/foundation-physics-graph-model
cd "$REPO" || exit 1

# CUDA is unusable on this host without the 570-series library shim.
source scripts/nvidia_lib_shim.sh >/dev/null 2>&1

# The GPUs are shared; take whichever currently has the most free memory.
GPU=$(nvidia-smi --query-gpu=index,memory.free --format=csv,noheader,nounits \
      | sort -t, -k2 -rn | head -1 | cut -d, -f1)

echo "[$(date '+%F %T')] TRELLIS.2 start | GPU=$GPU | simplify=20000 texture=1024"

CUDA_VISIBLE_DEVICES="$GPU" \
PYTHONPATH="$REPO/third_party/TRELLIS.2" \
HF_TOKEN="${HF_TOKEN:-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}" \
/home/quang/miniconda3/envs/trellis2/bin/python -u scripts/trellis_generate.py \
    --image /tmp/brick_512.png \
    --out "$REPO/outputs/trellis_brick.glb" \
    --resolution 512 \
    --simplify 20000 \
    --texture-size 1024
status=$?

echo "[$(date '+%F %T')] TRELLIS.2 exit=$status"
ls -la "$REPO/outputs/trellis_brick.glb" 2>/dev/null || echo "no GLB produced"
cat "$REPO/outputs/trellis_brick.json" 2>/dev/null