#!/usr/bin/env bash
# kromoss-image entrypoint. Runs inside the container; takes no arguments.
set -eu

# Weights are mounted read-only at /weights/kromoss; nothing is fetched at runtime.
# The custom pipeline (pipeline_moss_image.py) is loaded from that mount with
# trust_remote_code; HF caches the module copy under $HOME (/scratch).
export HF_HUB_OFFLINE=1
export TRANSFORMERS_OFFLINE=1
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

exec python3 -u /capsule/kromoss_server.py