m64_ea65644cdec9 / start.sh
12 lines, 495 Braw
1#!/usr/bin/env bash2# kromoss-dmd2-gb10 entrypoint. Runs inside the container; takes no arguments.3set -eu45# Weights are mounted read-only at /weights/kromoss; nothing is fetched at runtime.6# The custom pipeline (pipeline_moss_image.py) is loaded from that mount with7# trust_remote_code; HF caches the module copy under $HOME (/scratch).8export HF_HUB_OFFLINE=19export TRANSFORMERS_OFFLINE=110export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True1112exec python3 -u /capsule/kromoss_server.py