diff options
| author | Matthieu Pignolet <m@mpgn.dev> | 2025-01-29 13:35:05 +0400 |
|---|---|---|
| committer | Matthieu Pignolet <m@mpgn.dev> | 2025-01-29 13:35:05 +0400 |
| commit | 3d5c4980a17b7635c34393006a5cc15ab128cec0 (patch) | |
| tree | 08f498c382fb8447c4e86f769baecd82fa2690db | |
| parent | 7d0ac4cfb84abd99b53c8d0f1268157dca936364 (diff) | |
use gpu in deep phonemizer
| -rw-r--r-- | docker-compose.yaml | 7 |
1 files changed, 7 insertions, 0 deletions
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 |