selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-21 04:39:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

separate requirements

Browse Source
This commit is contained in:
Gokul Mohanarangan
2023-07-25 13:47:23 +05:30
parent d96c9d6adf
commit ef9a6a2e66
4 changed files with 84 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
scripts/setup_dependencies.sh
View File

@@ -1,33 +1,4 @@
#!/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
pip install -r ../requirements.txt
# download spacy models
spacy download en_core_web_sm
spacy download en_core_web_md
pip install -r ../server-requirements.txt