mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-20 20:29:06 +00:00
27 lines
546 B
Bash
Executable File
27 lines
546 B
Bash
Executable File
# 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
|
|
|