Update startup.sh

startup.sh

@@ -4,36 +4,54 @@
 
 set -Eeuo pipefail
 
-# ── CUDA allocator & MatAnyone scaling (fix fragmentation / VRAM pressure) ──
+# ── CUDA allocator & MatAnyone scaling (reduce fragmentation / VRAM pressure) ──
 export PYTORCH_CUDA_ALLOC_CONF="${PYTORCH_CUDA_ALLOC_CONF:-expandable_segments:True,max_split_size_mb:128}"
+export CUDA_MODULE_LOADING="${CUDA_MODULE_LOADING:-LAZY}"
+
+# MatAnyone loader/adapter hints (these are read in MatAnyoneLoader + CoreVideoProcessor)
 export MATANYONE_MAX_EDGE="${MATANYONE_MAX_EDGE:-640}"
 export MATANYONE_TARGET_PIXELS="${MATANYONE_TARGET_PIXELS:-400000}"
 
+# Windowed two-phase pipeline (SAM2 window → release → MatAnyone window)
+export MATANYONE_WINDOWED="${MATANYONE_WINDOWED:-1}"
+export MATANYONE_WINDOW="${MATANYONE_WINDOW:-8}"
+
+# Model-only downscale cap (applied in CoreVideoProcessor)
+export MAX_MODEL_SIZE="${MAX_MODEL_SIZE:-1280}"
+
 # ── Threads / general ──
 export OMP_NUM_THREADS="${OMP_NUM_THREADS:-4}"
 export MKL_NUM_THREADS="${MKL_NUM_THREADS:-4}"
 export PYTHONUNBUFFERED=1
 
+# ── Hugging Face caches (keeps downloads local to repo; avoids symlinks in some envs) ──
+export HF_HOME="${HF_HOME:-$PWD/checkpoints/hf}"
+export TRANSFORMERS_CACHE="${TRANSFORMERS_CACHE:-$HF_HOME}"
+export HF_DATASETS_CACHE="${HF_DATASETS_CACHE:-$HF_HOME}"
+export HF_HUB_DISABLE_SYMLINKS="${HF_HUB_DISABLE_SYMLINKS:-1}"
+export HF_HUB_ENABLE_HF_TRANSFER="${HF_HUB_ENABLE_HF_TRANSFER:-0}"
+
 # ── Gradio ──
 export GRADIO_SERVER_NAME="${GRADIO_SERVER_NAME:-0.0.0.0}"
 export GRADIO_SERVER_PORT="${GRADIO_SERVER_PORT:-7860}"
 
-# ── Hugging Face caches (keeps downloads local to repo) ──
-export HF_HOME="${HF_HOME:-$PWD/checkpoints/hf}"
-export TRANSFORMERS_CACHE="${TRANSFORMERS_CACHE:-$HF_HOME}"
-
 # ── Banner ──
 echo "===== BackgroundFX Pro Starting ====="
 echo "OMP_NUM_THREADS=$OMP_NUM_THREADS"
 echo "MKL_NUM_THREADS=$MKL_NUM_THREADS"
 echo "PYTORCH_CUDA_ALLOC_CONF=$PYTORCH_CUDA_ALLOC_CONF"
+echo "CUDA_MODULE_LOADING=$CUDA_MODULE_LOADING"
 echo "MATANYONE_MAX_EDGE=$MATANYONE_MAX_EDGE"
 echo "MATANYONE_TARGET_PIXELS=$MATANYONE_TARGET_PIXELS"
+echo "MATANYONE_WINDOWED=$MATANYONE_WINDOWED"
+echo "MATANYONE_WINDOW=$MATANYONE_WINDOW"
+echo "MAX_MODEL_SIZE=$MAX_MODEL_SIZE"
 echo "HF_HOME=$HF_HOME"
+echo "HF_HUB_DISABLE_SYMLINKS=$HF_HUB_DISABLE_SYMLINKS"
 echo "GRADIO_SERVER_NAME=$GRADIO_SERVER_NAME"
 echo "GRADIO_SERVER_PORT=$GRADIO_SERVER_PORT"
 echo "====================================="
 command -v nvidia-smi >/dev/null 2>&1 && nvidia-smi || true
 
 # ── Launch ──
-exec python app.py
+exec python -X faulthandler app.py