diff --git a/README.md b/README.md index ba211ee..d28b008 100644 --- a/README.md +++ b/README.md @@ -16,6 +16,9 @@ git clone https://github.com/shirayu/whisper_streaming.git cd whisper_streaming poetry install --only main +# If you use GPU with CUDA 11.6 +poetry run pip install 'torch==1.12.0+cu116' -f https://download.pytorch.org/whl/torch_stable.html + # Run in English poetry run whisper_streaming --language en --model base -n 20 ```