mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-16 00:13:07 +00:00
Brandon Thomas
41 lines
1.1 KiB
Bash
Executable File
41 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -e
|
|
|
|
INSTALL_OPTION=${1:-"dev"}
|
|
|
|
|
|
PIP=pip
|
|
|
|
echo 'Uninstalling stuff'
|
|
${PIP} uninstall -y nemo_toolkit
|
|
${PIP} uninstall -y sacrebleu
|
|
|
|
# Kept for legacy purposes
|
|
${PIP} uninstall -y nemo_asr
|
|
${PIP} uninstall -y nemo_nlp
|
|
${PIP} uninstall -y nemo_tts
|
|
${PIP} uninstall -y nemo_simple_gan
|
|
${PIP} uninstall -y nemo_cv
|
|
|
|
${PIP} install -U setuptools
|
|
|
|
echo 'Installing nemo and nemo_text_processing'
|
|
if [[ "$INSTALL_OPTION" == "dev" ]]; then
|
|
${PIP} install --editable ".[all]"
|
|
else
|
|
rm -rf dist/
|
|
python setup.py bdist_wheel
|
|
DIST_FILE=$(find ./dist -name "*.whl" | head -n 1)
|
|
${PIP} install "${DIST_FILE}[all]"
|
|
fi
|
|
|
|
echo 'Installing additional nemo_text_processing conda dependency'
|
|
bash nemo_text_processing/setup.sh > /dev/null 2>&1 && echo "nemo_text_processing installed!" || echo "nemo_text_processing could not be installed!"
|
|
|
|
if [ -x "$(command -v conda)" ]; then
|
|
echo 'Attempting update to numba installation via conda'
|
|
conda update -c numba numba -y > /dev/null 2>&1 && echo "Numba updated!" || echo "Numba could not be updated!"
|
|
fi
|
|
|
|
echo 'All done!'
|