From 0923b76d7900fa5fd69313054bfdad8dfb591ff4 Mon Sep 17 00:00:00 2001 From: SWivid Date: Thu, 13 Feb 2025 02:07:24 +0800 Subject: [PATCH] update README.md, add nvidia device gradio infer docker compose file example --- README.md | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/README.md b/README.md index aea01f9..c02f36a 100644 --- a/README.md +++ b/README.md @@ -87,6 +87,33 @@ f5-tts_infer-gradio --port 7860 --host 0.0.0.0 f5-tts_infer-gradio --share ``` +
+NVIDIA device docker compose file example + +``` yaml +services: + f5-tts: + image: ghcr.io/swivid/f5-tts:main + ports: + - "7860:7860" + environment: + GRADIO_SERVER_PORT: 7860 + entrypoint: ["f5-tts_infer-gradio", "--port", "7860", "--host", "0.0.0.0"] + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: [gpu] + +volumes: + f5-tts: + driver: local +``` + +
+ ### 2. CLI Inference ```bash