mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-04-13 09:39:17 +00:00
-Add default voice code setting and update language code resolution logic
This commit is contained in:
parent
9b76ce2071
commit
24b31ccbb5
3 changed files with 26 additions and 16 deletions
|
@ -13,6 +13,7 @@ class Settings(BaseSettings):
|
|||
output_dir: str = "output"
|
||||
output_dir_size_limit_mb: float = 500.0 # Maximum size of output directory in MB
|
||||
default_voice: str = "af_heart"
|
||||
default_voice_code: str | None = None # If set, overrides the first letter of voice name, though api call param still takes precedence
|
||||
use_gpu: bool = True # Whether to use GPU acceleration if available
|
||||
allow_local_voice_saving: bool = (
|
||||
False # Whether to allow saving combined voices locally
|
||||
|
|
|
@ -139,8 +139,14 @@ class KokoroV1(BaseModelBackend):
|
|||
await paths.save_voice_tensor(voice_tensor, temp_path)
|
||||
voice_path = temp_path
|
||||
|
||||
# Use provided lang_code or get from voice name
|
||||
pipeline_lang_code = lang_code if lang_code else voice_name[0].lower()
|
||||
# Use provided lang_code, settings voice code override, or first letter of voice name
|
||||
if lang_code: # api is given priority
|
||||
pipeline_lang_code = lang_code
|
||||
elif settings.default_voice_code: # settings is next priority
|
||||
pipeline_lang_code = settings.default_voice_code
|
||||
else: # voice name is default/fallback
|
||||
pipeline_lang_code = voice_name[0].lower()
|
||||
|
||||
pipeline = self._get_pipeline(pipeline_lang_code)
|
||||
|
||||
logger.debug(
|
||||
|
@ -232,8 +238,8 @@ class KokoroV1(BaseModelBackend):
|
|||
await paths.save_voice_tensor(voice_tensor, temp_path)
|
||||
voice_path = temp_path
|
||||
|
||||
# Use provided lang_code or get from voice name
|
||||
pipeline_lang_code = lang_code if lang_code else voice_name[0].lower()
|
||||
# Use provided lang_code, settings voice code override, or first letter of voice name
|
||||
pipeline_lang_code = lang_code if lang_code else (settings.default_voice_code if settings.default_voice_code else voice_name[0].lower())
|
||||
pipeline = self._get_pipeline(pipeline_lang_code)
|
||||
|
||||
logger.debug(
|
||||
|
|
|
@ -12,6 +12,7 @@ RUN apt-get update && apt-get install -y \
|
|||
libsndfile1 \
|
||||
curl \
|
||||
ffmpeg \
|
||||
g++ \
|
||||
&& apt-get clean && rm -rf /var/lib/apt/lists/* \
|
||||
&& mkdir -p /usr/share/espeak-ng-data \
|
||||
&& ln -s /usr/lib/*/espeak-ng-data/* /usr/share/espeak-ng-data/
|
||||
|
@ -19,8 +20,10 @@ RUN apt-get update && apt-get install -y \
|
|||
# Install UV using the installer script
|
||||
RUN curl -LsSf https://astral.sh/uv/install.sh | sh && \
|
||||
mv /root/.local/bin/uv /usr/local/bin/ && \
|
||||
mv /root/.local/bin/uvx /usr/local/bin/ && \
|
||||
useradd -m -u 1000 appuser && \
|
||||
mv /root/.local/bin/uvx /usr/local/bin/
|
||||
|
||||
# Create non-root user and set up directories and permissions
|
||||
RUN useradd -m -u 1000 appuser && \
|
||||
mkdir -p /app/api/src/models/v1_0 && \
|
||||
chown -R appuser:appuser /app
|
||||
|
||||
|
@ -30,28 +33,28 @@ WORKDIR /app
|
|||
# Copy dependency files
|
||||
COPY --chown=appuser:appuser pyproject.toml ./pyproject.toml
|
||||
|
||||
ENV PHONEMIZER_ESPEAK_PATH=/usr/bin \
|
||||
PHONEMIZER_ESPEAK_DATA=/usr/share/espeak-ng-data \
|
||||
ESPEAK_DATA_PATH=/usr/share/espeak-ng-data
|
||||
|
||||
# Install dependencies with GPU extras (using cache mounts)
|
||||
RUN --mount=type=cache,target=/root/.cache/uv \
|
||||
uv venv && \
|
||||
uv sync --extra gpu
|
||||
|
||||
# Copy project files including models and sync again
|
||||
# Copy project files including models
|
||||
COPY --chown=appuser:appuser api ./api
|
||||
COPY --chown=appuser:appuser web ./web
|
||||
COPY --chown=appuser:appuser docker/scripts/ ./
|
||||
RUN chmod +x ./entrypoint.sh
|
||||
RUN --mount=type=cache,target=/root/.cache/uv \
|
||||
uv sync --extra gpu
|
||||
|
||||
|
||||
# Set all environment variables in one go
|
||||
ENV PYTHONUNBUFFERED=1 \
|
||||
PYTHONPATH=/app:/app/api \
|
||||
PATH="/app/.venv/bin:$PATH" \
|
||||
UV_LINK_MODE=copy \
|
||||
USE_GPU=true \
|
||||
PHONEMIZER_ESPEAK_PATH=/usr/bin \
|
||||
PHONEMIZER_ESPEAK_DATA=/usr/share/espeak-ng-data \
|
||||
ESPEAK_DATA_PATH=/usr/share/espeak-ng-data
|
||||
USE_GPU=true
|
||||
|
||||
ENV DOWNLOAD_MODEL=true
|
||||
# Download model if enabled
|
||||
|
|
Loading…
Add table
Reference in a new issue