File size: 435 Bytes
4719196 | 1 2 3 4 5 6 7 8 9 | #!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "$0")/.."
ENV="${BRAINMU_ENV:-$PWD/.venv}"
[[ -x "$ENV/bin/python" ]] || { echo "Run bash scripts/setup_uv.sh first, or set BRAINMU_ENV to an existing compatible environment."; exit 2; }
export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0}"
export TOKENIZERS_PARALLELISM=false
exec uv run --no-project --python "$ENV/bin/python" python scripts/infer.py --expected-count 1000 "$@"
|