Update startup.sh

startup.sh

@@ -1,21 +1,39 @@
-#!/bin/bash
+#!/usr/bin/env bash
 # startup.sh - BackgroundFX Pro startup script
-# Sets environment variables before Python starts to avoid libgomp errors
+# Ensure all env vars are set BEFORE Python imports torch.
 
-# Set OpenMP threads BEFORE Python starts
-export OMP_NUM_THREADS=4
-export MKL_NUM_THREADS=4
+set -Eeuo pipefail
 
-# Optional: Set other environment variables
+# ── CUDA allocator & MatAnyone scaling (fix fragmentation / VRAM pressure) ──
+export PYTORCH_CUDA_ALLOC_CONF="${PYTORCH_CUDA_ALLOC_CONF:-expandable_segments:True,max_split_size_mb:128}"
+export MATANYONE_MAX_EDGE="${MATANYONE_MAX_EDGE:-640}"
+export MATANYONE_TARGET_PIXELS="${MATANYONE_TARGET_PIXELS:-400000}"
+
+# ── Threads / general ──
+export OMP_NUM_THREADS="${OMP_NUM_THREADS:-4}"
+export MKL_NUM_THREADS="${MKL_NUM_THREADS:-4}"
 export PYTHONUNBUFFERED=1
-export GRADIO_SERVER_NAME="0.0.0.0"
-export GRADIO_SERVER_PORT="7860"
 
-# Log the environment settings
+# ── 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 "================================="
+echo "PYTORCH_CUDA_ALLOC_CONF=$PYTORCH_CUDA_ALLOC_CONF"
+echo "MATANYONE_MAX_EDGE=$MATANYONE_MAX_EDGE"
+echo "MATANYONE_TARGET_PIXELS=$MATANYONE_TARGET_PIXELS"
+echo "HF_HOME=$HF_HOME"
+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
 
-# Start the application
-python app.py
+# ── Launch ──
+exec python app.py