Make whisper use cuda

edurenye · Oct 27, 2023 · 592de08 · 592de08
1 parent fe4ddc4
commit 592de08
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docker-compose.gpu.yml b/docker-compose.gpu.yml
@@ -27,6 +27,7 @@ services:
             - driver: nvidia
               count: 1
               capabilities: [gpu]
+    command: [ "--model", "tiny-int8", "--language", "en", "--device", "cuda" ]
 
   wyoming-openwakeword:
     extends: