diff --git a/setup-venv-local-build.sh b/setup-venv-local-build.sh index 7354f27..1929759 100755 --- a/setup-venv-local-build.sh +++ b/setup-venv-local-build.sh @@ -57,6 +57,7 @@ fi echo "==> upgrading pip + build tools" "${VENV}/bin/pip" install --upgrade pip wheel setuptools pybind11 --quiet +"${VENV}/bin/pip" install torch # --- clone (skipped if already done) --- if [ ! -d "${BUILD_DIR}/src/.git" ]; then diff --git a/setup-venv.sh b/setup-venv.sh index 6e9affe..19b288c 100755 --- a/setup-venv.sh +++ b/setup-venv.sh @@ -23,9 +23,9 @@ if [ ! -d "${VENV}" ]; then python3 -m venv "${VENV}" fi -echo "==> installing faster-whisper" +echo "==> installing torch and faster-whisper" "${VENV}/bin/pip" install --upgrade pip --quiet -"${VENV}/bin/pip" install faster-whisper +"${VENV}/bin/pip" install torch faster-whisper echo "" echo "==> done. Venv ready at ${VENV}"