taomate-digital-human / scripts /inference /start_dialogue_server.sh
multimodalart's picture
multimodalart HF Staff
Upload folder using huggingface_hub
12e0009 verified
Raw
History Blame Contribute Delete
978 Bytes
#!/usr/bin/env bash
set -euo pipefail
DIALOGUE_HOST="${DIALOGUE_HOST:-127.0.0.1}"
DIALOGUE_PORT="${DIALOGUE_PORT:-7864}"
DIALOGUE_MODEL_NAME="${DIALOGUE_MODEL_NAME:-taolive-dialogue}"
DIALOGUE_CONTEXT_SIZE="${DIALOGUE_CONTEXT_SIZE:-4096}"
DIALOGUE_GPU="${DIALOGUE_GPU:-3}"
LLAMA_SERVER_BIN="${LLAMA_SERVER_BIN:-}"
DIALOGUE_MODEL_PATH="${DIALOGUE_MODEL_PATH:-}"
if [ ! -x "${LLAMA_SERVER_BIN}" ]; then
echo "LLAMA_SERVER_BIN must point to an executable llama-server." >&2
exit 2
fi
if [ ! -f "${DIALOGUE_MODEL_PATH}" ]; then
echo "DIALOGUE_MODEL_PATH must point to a GGUF dialogue model." >&2
exit 2
fi
exec env CUDA_VISIBLE_DEVICES="${DIALOGUE_GPU}" "${LLAMA_SERVER_BIN}" \
-m "${DIALOGUE_MODEL_PATH}" \
--host "${DIALOGUE_HOST}" \
--port "${DIALOGUE_PORT}" \
--alias "${DIALOGUE_MODEL_NAME}" \
--ctx-size "${DIALOGUE_CONTEXT_SIZE}" \
--parallel 1 \
--gpu-layers all \
--main-gpu 0 \
--split-mode none \
--no-webui \
--jinja \
--reasoning off