diff --git a/README.md b/README.md index 8bd177a..61e4f77 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -Wyoming Whisper SST docker container with Nvidia GPU support for Home-Assistant +Wyoming Whisper STT docker container with Nvidia GPU support for Home-Assistant https://github.com/rhasspy/wyoming-faster-whisper diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100644 index 0000000..5fc0e13 --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,25 @@ +services: + wyoming-whisper: + image: slackr31337/wyoming-whisper-gpu:latest + container_name: wyoming-whisper + environment: + - MODEL=base-int8 + - LANGUAGE=en + - COMPUTE_TYPE=int8 + - BEAM_SIZE=5 + ports: + - 10300:10300 + volumes: + - /path/to/persistent/data:/data + restart: unless-stopped + runtime: nvidia + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: 1 + capabilities: + - gpu + - utility + - compute \ No newline at end of file