mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-18 07:41:03 +00:00
Introduce code from Rust monorepo. This is a mess
This commit is contained in:
Executable
+40
@@ -0,0 +1,40 @@
|
||||
#!/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!'
|
||||
Reference in New Issue
Block a user