0xarchit commited on
Commit
30426eb
·
1 Parent(s): 3d3f603

Support CONTEXT_LENGTH env var (default 4096) for ctx-size

Browse files
Files changed (1) hide show
  1. start.sh +18 -2
start.sh CHANGED
@@ -10,8 +10,23 @@ export NUMEXPR_NUM_THREADS="$THREADS"
10
  export HF_HOME="${HF_HOME:-/data/hf-cache}"
11
  export MODEL_DIR="${MODEL_DIR:-/data/models}"
12
  export MODEL_PATH="${MODEL_PATH:-/data/models/model.gguf}"
 
13
 
14
- python3 /app/download_model.py
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
 
16
  LLAMA_SERVER_BIN="${LLAMA_SERVER_BIN:-/usr/local/bin/llama-server}"
17
 
@@ -36,4 +51,5 @@ if [[ -n "${PARALLEL:-}" ]]; then
36
  server_args+=(--parallel "$PARALLEL")
37
  fi
38
 
39
- exec "$LLAMA_SERVER_BIN" "${server_args[@]}"
 
 
10
  export HF_HOME="${HF_HOME:-/data/hf-cache}"
11
  export MODEL_DIR="${MODEL_DIR:-/data/models}"
12
  export MODEL_PATH="${MODEL_PATH:-/data/models/model.gguf}"
13
+ export CTX_SIZE="${CONTEXT_LENGTH:-${CTX_SIZE:-4096}}"
14
 
15
+ # Ensure data directories exist and are writable. Run as root in container startup.
16
+ mkdir -p "$HF_HOME" "$MODEL_DIR"
17
+ # try to chown to appuser; if that fails (readonly mount), relax permissions
18
+ if chown -R 65532:65532 "$HF_HOME" "$MODEL_DIR" 2>/dev/null; then
19
+ true
20
+ else
21
+ chmod -R a+rwx "$HF_HOME" "$MODEL_DIR" || true
22
+ fi
23
+
24
+ # Run downloader as the non-root `appuser` using the venv python when available.
25
+ if [ -x "/opt/venv/bin/python" ]; then
26
+ su -s /bin/sh appuser -c "/opt/venv/bin/python /app/download_model.py"
27
+ else
28
+ su -s /bin/sh appuser -c "python3 /app/download_model.py"
29
+ fi
30
 
31
  LLAMA_SERVER_BIN="${LLAMA_SERVER_BIN:-/usr/local/bin/llama-server}"
32
 
 
51
  server_args+=(--parallel "$PARALLEL")
52
  fi
53
 
54
+ server_cmd=("$LLAMA_SERVER_BIN" "${server_args[@]}")
55
+ exec su -s /bin/sh appuser -c "$(printf '%q ' "${server_cmd[@]}")"