mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-08-05 16:48:53 +00:00
Add start-gpu.sh script for GPU-enabled FastAPI deployment
- Create GPU-specific startup script - Set environment variables for GPU and project configuration - Use uv to install GPU extras and run FastAPI server
This commit is contained in:
parent
6e79b252d0
commit
70c0d506de
1 changed files with 16 additions and 0 deletions
16
start-gpu.sh
Executable file
16
start-gpu.sh
Executable file
|
@ -0,0 +1,16 @@
|
|||
#!/bin/bash
|
||||
|
||||
# Get project root directory
|
||||
PROJECT_ROOT=$(pwd)
|
||||
|
||||
# Set environment variables
|
||||
export USE_GPU=true
|
||||
export USE_ONNX=false
|
||||
export PYTHONPATH=$PROJECT_ROOT:$PROJECT_ROOT/api
|
||||
export MODEL_DIR=src/models
|
||||
export VOICES_DIR=src/voices/v1_0
|
||||
export WEB_PLAYER_PATH=$PROJECT_ROOT/web
|
||||
|
||||
# Run FastAPI with GPU extras using uv run
|
||||
uv pip install -e ".[gpu]"
|
||||
uv run --no-sync uvicorn api.src.main:app --host 0.0.0.0 --port 50888
|
Loading…
Add table
Reference in a new issue