From 3d5c4980a17b7635c34393006a5cc15ab128cec0 Mon Sep 17 00:00:00 2001
From: Matthieu Pignolet <m@mpgn.dev>
Date: Wed, 29 Jan 2025 13:35:05 +0400
Subject: [PATCH] use gpu in deep phonemizer

---
 docker-compose.yaml | 7 +++++++
 1 file changed, 7 insertions(+)

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
-- 
2.39.5