From: Matthieu Pignolet <m@mpgn.dev>
Date: Wed, 29 Jan 2025 09:35:05 +0000 (+0400)
Subject: use gpu in deep phonemizer
X-Git-Url: https://git.puffer.fish/?a=commitdiff_plain;h=3d5c4980a17b7635c34393006a5cc15ab128cec0;p=matthieu%2Fgru.git

use gpu in deep phonemizer
---

diff --git a/docker-compose.yaml b/docker-compose.yaml
index af14ad6..f040bc3 100644
--- 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