Create start.sh
Browse files
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[@]}"
|