46 lines
1.4 KiB
Bash
Executable File
46 lines
1.4 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
SIA_INSTALL_NO_NOTEBOOK=1
|
|
#SIA_INSTALL_NO_CORE=1
|
|
SIA_INSTALL_NO_ITB=1
|
|
SIA_INSTALL_NO_MISTRAL_INFER=1
|
|
#SIA_INSTALL_NO_GEMMA_INFER=1
|
|
SIA_INSTALL_NO_TRAIN=1
|
|
|
|
if [ -z "${SIA_INSTALL_NO_NOTEBOOK}" ]; then
|
|
echo "Installing venv for running notebooks..."
|
|
python3 -m venv /root/venvs/notebook
|
|
/root/venvs/notebook/bin/pip install jupyter ipykernel ipywidgets
|
|
/root/venvs/notebook/bin/ipython kernel install --name=notebook
|
|
fi
|
|
|
|
if [ -z "${SIA_INSTALL_NO_CORE}" ]; then
|
|
echo "Installing SIA core..."
|
|
python3 -m venv /root/venvs/sia
|
|
/root/venvs/sia/bin/pip install -e /root/sia
|
|
fi
|
|
|
|
if [ -z "${SIA_INSTALL_NO_ITB}" ]; then
|
|
echo "Installing ITB tool..."
|
|
python3 -m venv /root/venvs/itb
|
|
/root/venvs/itb/bin/pip install -e /root/sia/tools/itb
|
|
fi
|
|
|
|
if [ -z "${SIA_INSTALL_NO_MISTRAL_INFER}" ]; then
|
|
echo "Installing venv for mistral inference..."
|
|
python3 -m venv /root/venvs/mistral_infer
|
|
/root/venvs/mistral_infer/bin/pip install -e /root/sia/tools/mistral_infer
|
|
fi
|
|
|
|
if [ -z "${SIA_INSTALL_NO_GEMMA_INFER}" ]; then
|
|
echo "Installing venv for gemma inference..."
|
|
python3 -m venv /root/venvs/gemma_infer
|
|
/root/venvs/gemma_infer/bin/pip install -e /root/sia/tools/gemma_infer
|
|
fi
|
|
|
|
if [ -z "${SIA_INSTALL_NO_TRAIN}" ]; then
|
|
echo "Installing Train tool..."
|
|
python3 -m venv /root/venvs/train
|
|
/root/venvs/train/bin/pip install -e /root/sia/tools/train
|
|
/root/venvs/train/bin/ipython kernel install --name=train
|
|
fi |