oki602960 commited on
Commit
6f46d49
·
verified ·
1 Parent(s): eaddb89

Create start.sh

Browse files
Files changed (1) hide show
  1. start.sh +21 -0
start.sh ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env bash
2
+ set -euo pipefail
3
+
4
+ API_KEY_ARG=()
5
+ if [[ -n "${API_KEY:-}" ]]; then
6
+ API_KEY_ARG=(--api_key "${API_KEY}")
7
+ fi
8
+
9
+ THREADS_ARG=()
10
+ if [[ -n "${N_THREADS:-}" ]]; then
11
+ THREADS_ARG=(--n_threads "${N_THREADS}")
12
+ fi
13
+
14
+ exec python -m llama_cpp.server \
15
+ --hf_model_repo_id "${HF_MODEL_REPO_ID}" \
16
+ --model "${HF_MODEL_FILE}" \
17
+ --host 0.0.0.0 \
18
+ --port "${PORT}" \
19
+ --n_ctx "${N_CTX}" \
20
+ "${API_KEY_ARG[@]}" \
21
+ "${THREADS_ARG[@]}"