添加 NV A100 Patched 镜像合并/entrypoint.sh
This commit is contained in:
14
NV A100 Patched 镜像合并/entrypoint.sh
Normal file
14
NV A100 Patched 镜像合并/entrypoint.sh
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
MODEL_DIR=${MODEL_DIR:-/model}
|
||||||
|
FIX_TOKENIZER_DIR=/tmp/fixed_tokenizer
|
||||||
|
|
||||||
|
echo "[entrypoint] fixing tokenizer..."
|
||||||
|
python /opt/fix_tokenizer.py
|
||||||
|
|
||||||
|
echo "[entrypoint] starting vllm..."
|
||||||
|
|
||||||
|
exec vllm serve "$MODEL_DIR" \
|
||||||
|
--tokenizer "$FIX_TOKENIZER_DIR" \
|
||||||
|
"$@"
|
||||||
Reference in New Issue
Block a user