Gemma notebook working on runpod

scripts/bootstrap.sh

@@ -37,7 +37,7 @@ apt-get install -y \
 
 # Install Chrome
 if [ -z "${SIA_INSTALL_NO_ITB}" ]; then
-    wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add -
+    wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | gpg --dearmor -o /etc/apt/trusted.gpg.d/google.gpg
     echo "deb http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google-chrome.list
     apt-get update && \
         apt-get install -y \

scripts/collect.sh

@@ -11,6 +11,7 @@ declare -A FILTER_SETS=(
   
   ["core"]="-s py ./sia ./tools -s deploy . -f ^(?!procedures/).*\\.md$ ."
   ["lib"]="-s rust . -s py . -s doc . -s deploy ."
+  ["notebook"]="-f Dockerfile . -f scripts/install.sh . -f scripts/local.sh ."
   ["webui"]="-s web ./web"
   ["tests"]="-s py ./test"
   ["procedures"]="-s doc ./procedures"

scripts/install.sh

@@ -13,16 +13,16 @@ if [ -z "${SIA_INSTALL_NO_TRAIN}" ]; then
     /root/venvs/train/bin/ipython kernel install --name=train
 fi
 
-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
     /root/venvs/sia/bin/ipython kernel install --name=sia
+fi
+
+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

scripts/local.sh

@@ -33,6 +33,7 @@ docker run \
     -v /$(pwd)/tasks/:/root/data/tasks/ \
     -v /$(pwd)/user/:/root/data/user/ \
     -v /$(pwd)/environment/:/root/data/environment/ \
+    -v /$(pwd)/cache/:/root/.cache/ \
     -v /$(pwd)/:/root/sia/ \
     sia "$@"