0.0.20

.gitignore

@@ -1,2 +1,3 @@
 apps.json
-kaggle.json
+kaggle.json
+oauth2.cfg

Dockerfile

@@ -1,7 +1,7 @@
-# build with: docker build . --tag sctg/roco-idefics3:0.0.19 --tag sctg/roco-idefics3:latest --push
+# build with: docker build . --tag sctg/roco-idefics3:0.0.20 --tag sctg/roco-idefics3:latest --push
 # run with 
-# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -e WANDB_API_KEY=wdkfjzfjz -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.19 bash -i /start.sh sleep infinity
-# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -it sctg/roco-idefics3:0.0.19 bash -i /start.sh python /learn.py hf_...
+# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -e WANDB_API_KEY=wdkfjzfjz -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.20 bash -i /start.sh sleep infinity
+# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -it sctg/roco-idefics3:0.0.20 bash -i /start.sh python /learn.py hf_...
 FROM cloudflare/cloudflared:latest as cloudflared
 FROM nvidia/cuda:12.6.2-cudnn-devel-ubuntu22.04
 ARG NODE_MAJOR="20"
@@ -20,6 +20,7 @@ USER root
 RUN chown -R 42420:42420 /workspace
 RUN echo "ovh ALL=(ALL) NOPASSWD: ALL" >> /etc/sudoers
 COPY --from=cloudflared /usr/local/bin/cloudflared /usr/local/bin/cloudflared
+RUN curl -JL https://github.com/oauth2-proxy/oauth2-proxy/releases/download/v7.7.1/oauth2-proxy-v7.7.1.linux-amd64.tar.gz | tar -xz -C /usr/local/bin --strip-components=1
 USER 42420
 RUN curl -L https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh > /workspace/miniconda.sh
 COPY bitsandbytes-0.45.0-cp312-cp312-linux_x86_64.whl /tmp/bitsandbytes-0.45.0-cp312-cp312-linux_x86_64.whl
@@ -56,6 +57,7 @@ RUN mkdir -p /workspace/.config/github-copilot/
 COPY apps.json.enc /workspace/.config/github-copilot/apps.json.enc
 COPY kaggle.json.enc /workspace/.config/kaggle.json.enc
 COPY CFTOKEN.enc /workspace/.config/CFTOKEN.enc
+COPY oauth2.cfg.enc /workspace/.config/oauth2.cfg.enc
 # Mandatory to run the jobs in rootless mode
 # USER root
 # RUN chown -R 42420:42420 /workspace

oauth2.cfg.enc

@@ -0,0 +1,9 @@
+U2FsdGVkX19n4GPbJ9yrXIPRO6kl3oubEzskvhD3pWXrzhHUBeMSmxHGGHmy5D+K
+Px1eAnhHWVHj0DgeSdSJxZYhg/FWQUJX1uga0U2HDYJ7tgPFoHgHQOfabeSBBeq/
+SPLIhK4mDfERejvWICpUN5pWJYYWh0XtRpv2eCMVP6sIT6UtO+xVyJe2/+j5EyAY
+4oEv2QpFZEeF8pZa8w/QPrH1YWmWimTN9MHGZVlBk5x0eHKlYzq0clUEkD7Z5d5j
+QLHzEwxIxar+AGQE4i4itcE3ti4unbOCrijkb4zGinGwKBM46d3RbQoPWPf3xc27
+N3qkHIADGmFc+3zYgfUKvDjbrQjd/ZaMtSWX1DWnkifa3m/r5g6GMQv6DbdQkxIk
+/TimhTfYBhOPe5XOHiSB4gZKTxc/s7iD4lsWjgudxJISp+/yvzhyba1xWVOWTP+F
+u4tDY4ao8CX3hjUOWegAM5IWdQfUmgi648HRWVNmr/eCnz94qh1diDTP2LqYWDsC
+lQ3wvyUftKmJljSFWhrpjQ==

start.sh

@@ -45,6 +45,14 @@ if [ -f /workspace/.config/kaggle.json.enc ]; then
   openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" -in /workspace/.config/kaggle.json.enc -out /workspace/.kaggle/kaggle.json
 fi
 
+# Decode oauth2.cfg.enc file
+# encoded with: openssl aes-256-cbc -base64 -md sha256 -pass pass:"$HF_TOKEN" -in oauth2.cfg -out oauth2.cfg.enc
+# decode with: openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" -in oauth2.cfg.enc -out oauth2.cfg
+if [ -f /workspace/.config/oauth2.cfg.enc ]; then
+  openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" -in /workspace/.config/oauth2.cfg.enc -out /workspace/.config/oauth2.cfg
+  oauth2-proxy --config /workspace/.config/oauth2.cfg &
+fi
+
 git clone https://huggingface.co/eltorio/IDEFICS3_ROCOv2
 git clone https://huggingface.co/spaces/eltorio/Llama-3.2-3B-appreciation
 git config --global user.email "[email protected]"