public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-08-05 16:48:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Updated docker.

Browse source 
Removed unused py libs.
This commit is contained in:
faltiska 2025-06-12 08:20:45 +03:00
parent f18ab7dff7
commit 5425e537bc
3 changed files with 11 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
api/src/main.py
View file

@ -59,21 +59,15 @@ async def lifespan(app: FastAPI):
logger.error(f"Failed to initialize model: {e}")
raise
boundary = "░" * 2 * 12
startup_msg = f"""
{boundary}
╔═╗┌─┐┌─┐┌┬┐
╠╣ ├─┤└─┐ │
╚ ┴ ┴└─┘ ┴
╦╔═┌─┐┬┌─┌─┐
╠╩╗│ │├┴┐│ │
╩ ╩└─┘┴ ┴└─┘
{boundary}
"""
startup_msg += f"\nModel loaded on {device}"
startup_msg += f"Model loaded on {device}"
if device == "mps":
startup_msg += "\nUsing Apple Metal Performance Shaders (MPS)"
elif device == "cuda":

21
docker/gpu/Dockerfile
View file

@ -16,9 +16,7 @@ RUN apt-get update -y && \
USER appuser
WORKDIR /app
ENV PHONEMIZER_ESPEAK_PATH=/usr/bin \
PHONEMIZER_ESPEAK_DATA=/usr/share/espeak-ng-data \
PYTHONUNBUFFERED=1 \
ENV PYTHONUNBUFFERED=1 \
PYTHONPATH=/app:/app/api \
PATH="/app/.venv/bin:$PATH" \
UV_LINK_MODE=copy \
@ -26,15 +24,15 @@ ENV PHONEMIZER_ESPEAK_PATH=/usr/bin \
DEVICE="gpu"
COPY --chown=appuser:appuser pyproject.toml ./pyproject.toml
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 venv --python 3.10 && \
uv sync --extra gpu && \
uv cache clean && \
uv cache clean
COPY --chown=appuser:appuser api ./api
COPY --chown=appuser:appuser web ./web
COPY --chown=appuser:appuser docker/scripts/ ./
RUN chmod +x ./entrypoint.sh && \
python download_model.py --output api/src/models/v1_0
# Run FastAPI server through entrypoint.sh
@ -67,8 +65,9 @@ CMD ["./entrypoint.sh"]
# SSH keys for logging into EC2 containers:
# https://eu-west-1.console.aws.amazon.com/ec2/home?region=eu-west-1#KeyPairs:
#
# aws ecr get-login-password --region eu-west-1 | docker login --username AWS --password-stdin 678811077621.dkr.ecr.eu-west-1.amazonaws.com
# docker build -f docker\gpu\Dockerfile -t evie/kokorotts .
# aws ecr get-login-password --region eu-west-1 | docker login --username AWS --password-stdin 678811077621.dkr.ecr.eu-west-1.amazonaws.com
# docker tag evie/kokorotts:latest 678811077621.dkr.ecr.eu-west-1.amazonaws.com/evie/kokorotts:latest
# docker push 678811077621.dkr.ecr.eu-west-1.amazonaws.com/evie/kokorotts:latest
# docker run --gpus all -p 8880:8880 --name kokorotts 678811077621.dkr.ecr.eu-west-1.amazonaws.com/evie/kokorotts
# docker run --gpus all -p 8880:8880 --name kokorotts 678811077621.dkr.ecr.eu-west-1.amazonaws.com/evie/kokorotts
# oe use swarm as explained here: https://blog.container-solutions.com/rolling-updates-with-docker-swarm

2
pyproject.toml
View file

@ -12,7 +12,6 @@ dependencies = [
"pydantic==2.10.4",
"pydantic-settings==2.7.0",
"python-dotenv==1.0.1",
"sqlalchemy==2.0.27",
# ML/DL Base
"numpy>=1.26.0",
"scipy==1.14.1",
@ -28,7 +27,6 @@ dependencies = [
"loguru==0.7.3",
"openai>=1.59.6",
"pydub>=0.25.1",
"matplotlib>=3.10.0",
"mutagen>=1.47.0",
"psutil>=6.1.1",
"kokoro==0.9.2",