Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b50b645cc7 | ||
|
|
4cb1930e4a | ||
|
|
252d761afc |
17
vllm_wrapper.sh
Normal file
17
vllm_wrapper.sh
Normal file
@@ -0,0 +1,17 @@
|
||||
#!/bin/bash
|
||||
set -e
|
||||
|
||||
# 只拦截 "serve <model_dir>" 子命令,其他子命令直接透传
|
||||
if [ "$1" = "serve" ] && [ -n "$2" ]; then
|
||||
MODEL_DIR="$2"
|
||||
shift 2
|
||||
|
||||
FIXED_DIR=$(python3 /opt/fix_tokenizer.py "$MODEL_DIR")
|
||||
if [ -n "$FIXED_DIR" ]; then
|
||||
exec /opt/vllm_kunlun/bin/vllm_real serve "$MODEL_DIR" --tokenizer "$FIXED_DIR" "$@"
|
||||
else
|
||||
exec /opt/vllm_kunlun/bin/vllm_real serve "$MODEL_DIR" "$@"
|
||||
fi
|
||||
fi
|
||||
|
||||
exec /opt/vllm_kunlun/bin/vllm_real "$@"
|
||||
Reference in New Issue
Block a user