From 294ff694cb0a43fddca96ed0deaf2b6286947c82 Mon Sep 17 00:00:00 2001
From: Jimmy <git@jimmy.nz>
Date: Tue, 25 Oct 2022 22:15:01 +1300
Subject: [PATCH] Use gpu for vosk

---
 docker-compose.yaml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index 8b62df4..cdb7d2c 100644
--- 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
\ No newline at end of file
+      - 2700:2700
+    runtime: nvidia
\ No newline at end of file