mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-04-13 09:39:17 +00:00
32 lines
756 B
Bash
32 lines
756 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Ensure models directory exists
|
||
|
|
mkdir -p api/src/models
|
||
|
|
|
||
|
|
# Function to download a file
|
||
|
|
download_file() {
|
||
|
|
local url="$1"
|
||
|
|
local filename=$(basename "$url")
|
||
|
|
echo "Downloading $filename..."
|
||
|
|
curl -L "$url" -o "api/src/models/$filename"
|
||
|
|
}
|
||
|
|
|
||
|
|
# Default ONNX model if no arguments provided
|
||
|
|
DEFAULT_MODELS=(
|
||
|
|
"https://github.com/remsky/Kokoro-FastAPI/releases/download/v0.1.0/kokoro-v0_19.onnx"
|
||
|
|
"https://github.com/remsky/Kokoro-FastAPI/releases/download/v0.1.0/kokoro-v0_19_fp16.onnx"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Use provided models or default
|
||
|
|
if [ $# -gt 0 ]; then
|
||
|
|
MODELS=("$@")
|
||
|
|
else
|
||
|
|
MODELS=("${DEFAULT_MODELS[@]}")
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Download all models
|
||
|
|
for model in "${MODELS[@]}"; do
|
||
|
|
download_file "$model"
|
||
|
|
done
|
||
|
|
|
||
|
|
echo "ONNX model download complete!"
|