Use gpu for vosk

2022-10-25 22:15:01 +13:00 · 2022-10-25 22:15:01 +13:00 · 294ff694cb
parent 3a63a37777
commit 294ff694cb
1 changed files with 3 additions and 2 deletions
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@ -13,6 +13,7 @@ services:
      - ~/Downloads/sd-v1-4-full-ema.ckpt:/app/model.ckpt

  vosk:
-    image: alphacep/kaldi-en
+    image: alphacep/kaldi-en-gpu
    ports:
-      - 2700:2700
+      - 2700:2700
+    runtime: nvidia