mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-20 20:29:06 +00:00
40 lines
867 B
Bash
40 lines
867 B
Bash
#!/bin/sh
|
|
|
|
# Upgrade pip
|
|
pip install --upgrade pip
|
|
|
|
# Default to CPU Installation of JAX
|
|
jax_mode="jax[cpu]"
|
|
|
|
# Install JAX
|
|
if [ "$1" == "cpu" ]
|
|
then
|
|
jax_mode="jax[cpu]"
|
|
elif [ "$1" == "cuda11" ]
|
|
then
|
|
jax_mode="jax[cuda11_pip]"
|
|
elif [ "$1" == "cuda12" ]
|
|
then
|
|
jax_mode="jax[cuda12_pip]"
|
|
fi
|
|
|
|
pip install --upgrade "$jax_mode"
|
|
|
|
# Install Whisper-JAX base
|
|
pip install git+https://github.com/sanchit-gandhi/whisper-jax.git
|
|
|
|
# Update to latest version
|
|
pip install --upgrade --no-deps --force-reinstall git+https://github.com/sanchit-gandhi/whisper-jax.git
|
|
|
|
cwd=$(pwd)
|
|
last_component="${cwd##*/}"
|
|
if [ "$last_component" = "reflector" ]; then
|
|
pip install -r pipeline-requirements.txt
|
|
elif [ "$last_component" = "scripts" ]; then
|
|
pip install -r ../pipeline-requirements.txt
|
|
fi
|
|
|
|
# download spacy models
|
|
spacy download en_core_web_sm
|
|
spacy download en_core_web_md
|