use gpu in deep phonemizer

diff --git a/docker-compose.yaml b/docker-compose.yaml
index af14ad6f5e17faae6e3f7a20930366f6d409fc70..f040bc36243091c9ebe17e5ff2041c7ee975dc8e 100644 (file)
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -27,3 +27,10 @@ services:
       - 8000:8000
     volumes:
       - ./deep_phonemizer/assets/model.pt:/app/assets/model.pt
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
\ No newline at end of file