34 lines
924 B
Bash
34 lines
924 B
Bash
#!/bin/bash
|
|
cat /welcome.txt
|
|
export PYTHONUNBUFFERED=1
|
|
|
|
if [[ ! -f "v2-inference-v.yaml" ]]; then
|
|
python utils/get_yamls.py
|
|
fi
|
|
|
|
mkdir -p logs input
|
|
|
|
# RunPod SSH
|
|
if [[ -v "PUBLIC_KEY" ]] && [[ ! -d "${HOME}/.ssh" ]]
|
|
then
|
|
pushd $HOME
|
|
mkdir -p .ssh
|
|
echo ${PUBLIC_KEY} > .ssh/authorized_keys
|
|
chmod -R 700 .ssh
|
|
popd
|
|
service ssh start
|
|
fi
|
|
|
|
tensorboard --logdir /workspace/EveryDream2trainer/logs --host 0.0.0.0 &
|
|
|
|
# RunPod JupyterLab
|
|
if [[ $JUPYTER_PASSWORD ]]
|
|
then
|
|
jupyter nbextension enable --py widgetsnbextension
|
|
jupyter labextension disable "@jupyterlab/apputils-extension:announcements"
|
|
jupyter lab --allow-root --no-browser --port=8888 --ip=* --ServerApp.terminado_settings='{"shell_command":["/bin/bash"]}' --ServerApp.token=$JUPYTER_PASSWORD --ServerApp.allow_origin=* --ServerApp.preferred_dir=/workspace/EveryDream2trainer
|
|
else
|
|
echo "Container Started"
|
|
sleep infinity
|
|
fi
|