Files
pfm-ocr/backend/scripts/install.sh
T

29 lines
776 B
Bash

#!/usr/bin/env bash
set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$ROOT"
# Avoid picking up another project's active venv
unset VIRTUAL_ENV
PYTHON="${PYTHON:-3.12}"
FLASH_ATTN_WHEEL="${FLASH_ATTN_WHEEL:-}"
echo "==> Using uv with Python ${PYTHON}"
uv python pin "${PYTHON}"
uv sync
if [[ -n "${FLASH_ATTN_WHEEL}" ]]; then
echo "==> Installing prebuilt flash-attn wheel"
uv pip install --python .venv "${FLASH_ATTN_WHEEL}"
else
echo "==> Installing flash-attn (may require a prebuilt wheel; set FLASH_ATTN_WHEEL)"
uv pip install --python .venv "flash-attn==2.8.2"
fi
echo "==> Verifying paddleocr genai_server entrypoint"
uv run paddleocr genai_server --help >/dev/null
echo "==> Done. Start the service with: ./scripts/serve.sh"