File size: 963 Bytes
49f972e a1c753a 49f972e 44a9e20 0057964 66f8ee4 44a9e20 49f972e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
#!/bin/bash
echo "pod started"
if [[ $PUBLIC_KEY ]]
then
mkdir -p ~/.ssh
chmod 700 ~/.ssh
cd ~/.ssh
echo $PUBLIC_KEY >> authorized_keys
chmod 700 -R ~/.ssh
cd /
service ssh start
fi
#!/bin/bash
if [[ -z "${HF_TOKEN}" ]] || [[ "${HF_TOKEN}" == "enter_your_huggingface_token_here" ]]
then
echo "HF_TOKEN is not set"
else
echo "HF_TOKEN is set, logging in..."
huggingface-cli login --token ${HF_TOKEN}
fi
# Start nginx as reverse proxy to enable api access
service nginx start
# Start JupyterLab
jupyter lab --ip=0.0.0.0 --port=8888 --no-browser --allow-root --NotebookApp.allow_origin='*' --NotebookApp.token='' &
echo "JupyterLab started"
# Check if there is a venv directory, if so, activate it
if [ -d "/workspace/venv" ]; then
echo "venv directory found, activating it"
source /workspace/venv/bin/activate
fi
# RUN COMFYUI
python3 /workspace/ComfyUI/main.py --listen
/download_models.sh
sleep infinity
|