mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2025-12-20 20:29:06 +00:00
Moved all server files to server/
This commit is contained in:
39
server/scripts/setup_pipeline_dependencies.sh
Normal file
39
server/scripts/setup_pipeline_dependencies.sh
Normal file
@@ -0,0 +1,39 @@
|
||||
#!/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
|
||||
Reference in New Issue
Block a user