#!/usr/bin/env bash
# flux2-klein-4b-gb10 entrypoint. Runs inside the container; takes no arguments.
set -eu

# Weights are mounted read-only at /weights/flux2; nothing is fetched at runtime.
export HF_HUB_OFFLINE=1
export TRANSFORMERS_OFFLINE=1

export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True

exec python3 /capsule/flux2_server.py