Spaces:
Running on Zero
Running on Zero
File size: 2,056 Bytes
6143b69 afa0684 6143b69 afa0684 6143b69 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 | # Start the v2 RICS backend for local testing.
# Run from repo root: .\backend\scripts\start_server.ps1
$ErrorActionPreference = "Stop"
$RepoRoot = Split-Path (Split-Path $PSScriptRoot -Parent) -Parent
Set-Location $RepoRoot
$DataDir = Join-Path $RepoRoot ".rics_v2_data"
New-Item -ItemType Directory -Force -Path (Join-Path $DataDir "tmp") | Out-Null
$env:DATA_DIR = $DataDir
$env:MASTER_TEMPLATE_AUTO_INGEST = "false"
# Reference ingest follows .env (REFERENCE_AUTO_INGEST_ENABLED=true loads past reports).
# jina-reranker-v3 warmup off locally: lazy-loads on first reference mapping
# (small Windows paging file can hard-abort an eager startup load).
$env:REFERENCE_CROSS_ENCODER_WARMUP = "false"
$env:SPACY_MODEL = "en_core_web_sm"
# Use CUDA only when torch was built with GPU support; CPU-only torch crashes on device=cuda.
$cudaOk = & python -c "import torch; print('1' if torch.cuda.is_available() else '0')" 2>$null
if ($cudaOk -eq "1") {
$env:LOCAL_EMBEDDING_DEVICE = "cuda"
} else {
$env:LOCAL_EMBEDDING_DEVICE = "cpu"
Write-Host "LOCAL_EMBEDDING_DEVICE=cpu (no CUDA torch/GPU)"
}
$env:LOCAL_EMBEDDING_BATCH_SIZE = "4"
$env:LOCAL_EMBEDDING_DTYPE = "bfloat16"
# Cap parallel section workers (default 54 OOM-kills the process on 4 GB GPU / small paging file).
# 2 is safer for long 26–30 section runs on 16 GB RAM + small Windows paging file.
$env:SECTION_CONCURRENCY = "2"
$env:MAX_CONCURRENT_LLM_CALLS = "2"
# Regex-only PII during generation — spaCy NER + jina embedder + reranker exhaust host RAM mid-run.
$env:PII_USE_SPACY = "false"
# jina-reranker-v3 on CPU: it cannot share the 4 GB card with the resident embedder
# (native CUDA/driver hard-abort mid-generation). Embedder stays on cuda above.
$env:REFERENCE_CROSS_ENCODER_DEVICE = "cpu"
Write-Host "DATA_DIR=$($env:DATA_DIR)"
Write-Host "REFERENCE_CROSS_ENCODER_WARMUP=$($env:REFERENCE_CROSS_ENCODER_WARMUP)"
Write-Host "SPACY_MODEL=$($env:SPACY_MODEL)"
Write-Host "Starting http://127.0.0.1:8000 ..."
python -m uvicorn backend.main:app --host 127.0.0.1 --port 8000
|