| (return 0 2>/dev/null) || { |
| echo "!! setup_env.sh must be SOURCED, not executed:" |
| echo " source setup_env.sh" |
| exit 1 |
| } |
|
|
| if [ ! -d /workspace ]; then |
| echo "!! /workspace not found — attach a RunPod network volume at /workspace" |
| echo " (model weights, uv cache and DB backups all live there)." |
| return 1 |
| fi |
|
|
| if [ "$(id -u)" -eq 0 ]; then SUDO=""; else SUDO="sudo"; fi |
|
|
| export HF_HOME="${HF_HOME:-/workspace/hf_cache}" |
| export UV_CACHE_DIR="${UV_CACHE_DIR:-/workspace/.uv/cache}" |
| export UV_PYTHON_INSTALL_DIR="${UV_PYTHON_INSTALL_DIR:-/workspace/.uv/python}" |
| export UV_LINK_MODE="${UV_LINK_MODE:-copy}" |
| mkdir -p "$HF_HOME" "$UV_CACHE_DIR" "$UV_PYTHON_INSTALL_DIR" |
|
|
| if ! command -v curl >/dev/null 2>&1; then |
| echo ">> Installing curl/ca-certificates..." |
| $SUDO apt-get update -y |
| $SUDO apt-get install -y --no-install-recommends curl ca-certificates |
| fi |
|
|
| export PATH="$HOME/.local/bin:$PATH" |
| if ! command -v uv >/dev/null 2>&1; then |
| echo ">> Installing uv..." |
| curl -LsSf https://astral.sh/uv/install.sh | sh |
| else |
| echo ">> uv already installed ($(uv --version)) — skipping." |
| fi |
|
|
| export PROFILER_AGENT_PORT="${PROFILER_AGENT_PORT:-8011}" |
| export FORENSIC_AGENT_PORT="${FORENSIC_AGENT_PORT:-8002}" |
|
|
| if [ -z "${HF_HUB_OFFLINE+x}" ]; then |
| if ls -d "$HF_HOME"/hub/models--* >/dev/null 2>&1; then |
| export HF_HUB_OFFLINE=1 |
| else |
| echo ">> HF cache is empty; leaving HF_HUB_OFFLINE off so weights can download." |
| echo " (Set HF_TOKEN in .env first if either model repo is private/gated.)" |
| echo " Re-source this script after the first run to switch to offline mode." |
| fi |
| fi |
|
|
|
|
