e1f966ace3
* Update serge.env Add ISERGE_ENABLE_IPV4 * Update deploy.sh Now IPV4 is activate by deflault We can activate IPV4+IPV6 or IPV6 only * Update dev.sh Now ipv4 is activated by default but we can also activate ipv4+ipv6 or ipv6 only * Update dev.sh fix port for ipv4 * Update serge.env fix SERGE_ENABLE_IPV4 value * Update deploy.sh code formating * Update dev.sh code formating * Update dev.sh bugfix * Update serge.env --------- Co-authored-by: Juan Calderon-Perez <835733+gaby@users.noreply.github.com>
72 lines
1.9 KiB
Bash
Executable File
72 lines
1.9 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -x
|
|
source serge.env
|
|
|
|
# Get CPU Architecture
|
|
cpu_arch=$(uname -m)
|
|
|
|
# Function to detect CPU features
|
|
detect_cpu_features() {
|
|
cpu_info=$(lscpu)
|
|
if echo "$cpu_info" | grep -q "avx512"; then
|
|
echo "AVX512"
|
|
elif echo "$cpu_info" | grep -q "avx2"; then
|
|
echo "AVX2"
|
|
elif echo "$cpu_info" | grep -q "avx"; then
|
|
echo "AVX"
|
|
else
|
|
echo "basic"
|
|
fi
|
|
}
|
|
|
|
# Check if the CPU architecture is aarch64/arm64
|
|
if [ "$cpu_arch" = "aarch64" ]; then
|
|
pip_command="python -m pip install -v llama-cpp-python==$LLAMA_PYTHON_VERSION --only-binary=:all: --extra-index-url=https://gaby.github.io/arm64-wheels/"
|
|
else
|
|
# Use @smartappli provided wheels
|
|
cpu_feature=$(detect_cpu_features)
|
|
pip_command="python -m pip install -v llama-cpp-python==$LLAMA_PYTHON_VERSION --only-binary=:all: --extra-index-url=https://smartappli.github.io/serge-wheels/$cpu_feature/cpu"
|
|
fi
|
|
|
|
echo "Recommended install command for llama-cpp-python: $pip_command"
|
|
|
|
# Install python vendor dependencies
|
|
pip install -r /usr/src/app/requirements.txt || {
|
|
echo 'Failed to install python dependencies from requirements.txt'
|
|
exit 1
|
|
}
|
|
|
|
# Install python dependencies
|
|
pip install -e ./api || {
|
|
echo 'Failed to install python dependencies'
|
|
exit 1
|
|
}
|
|
|
|
# Install python bindings
|
|
eval "$pip_command" || {
|
|
echo 'Failed to install llama-cpp-python'
|
|
exit 1
|
|
}
|
|
|
|
# Start Redis instance
|
|
redis-server /etc/redis/redis.conf &
|
|
|
|
# Start the web server
|
|
cd /usr/src/app/web || exit 1
|
|
npm run dev -- --host 0.0.0.0 --port 8008 &
|
|
|
|
# Start the API
|
|
cd /usr/src/app/api || exit 1
|
|
hypercorn_cmd="hypercorn src.serge.main:api_app --bind 0.0.0.0:9124"
|
|
if [ "$SERGE_ENABLE_IPV6" = true ] && [ "$SERGE_ENABLE_IPV4" != true ]; then
|
|
hypercorn_cmd="hypercorn src.serge.main:api_app --bind [::]:9124"
|
|
elif [ "$SERGE_ENABLE_IPV4" = true ] && [ "$SERGE_ENABLE_IPV6" = true ]; then
|
|
hypercorn_cmd="hypercorn src.serge.main:api_app --bind 0.0.0.0:9124 --bind [::]:9124"
|
|
fi
|
|
|
|
$hypercorn_cmd || {
|
|
echo 'Failed to start main app'
|
|
exit 1
|
|
}
|