From 17f4e361911232c60b473898b3ea8a46c4e58781 Mon Sep 17 00:00:00 2001 From: Yuta Hayashibe Date: Sat, 24 Sep 2022 12:38:37 +0900 Subject: [PATCH] Fix instruction --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d28b008..1afd910 100644 --- a/README.md +++ b/README.md @@ -16,8 +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 +# If you use GPU, intall proper torch with "poetry run pip install -U" +# Example : torch for CUDA 11.6 +poetry run pip install -U torch --extra-index-url https://download.pytorch.org/whl/cu116 # Run in English poetry run whisper_streaming --language en --model base -n 20