matthieu/Gru
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

use gpu in deep phonemizer

Browse source
This commit is contained in:
matthieu 2025-01-29 13:35:05 +04:00
parent 7d0ac4cfb8
commit 3d5c4980a1
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
docker-compose.yaml
View file

@ -27,3 +27,10 @@ services:
- 8000:8000 - 8000:8000
volumes: volumes:
- ./deep_phonemizer/assets/model.pt:/app/assets/model.pt - ./deep_phonemizer/assets/model.pt:/app/assets/model.pt
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]