wyoming-whisper-gpu/run.sh
Confused-Engineer a7d5c90080
Update run.sh
2024-01-10 12:14:37 -05:00

10 lines
308 B
Bash

#!/usr/bin/env bash
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/python3.10/dist-packages/nvidia/cudnn/lib/
python3 -m wyoming_faster_whisper \
--uri 'tcp://0.0.0.0:10300' \
--data-dir /data \
--download-dir /data "$@" \
--model medium-int8 \
--beam-size 5 \
--device cuda