none
commited on
Commit
·
ebb4897
1
Parent(s):
e310801
Update cuda
Browse files- Dockerfile +4 -4
- start.sh +3 -1
Dockerfile
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
-
# build with: docker build . --tag sctg/roco-idefics3:0.0.
|
2 |
# run with
|
3 |
-
# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.
|
4 |
-
# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -it sctg/roco-idefics3:0.0.
|
5 |
FROM cloudflare/cloudflared:latest as cloudflared
|
6 |
-
FROM nvidia/cuda:
|
7 |
ARG NODE_MAJOR="20"
|
8 |
RUN /usr/sbin/addgroup --gid 42420 ovh
|
9 |
RUN /usr/sbin/useradd -u 42420 --gid 42420 -m -d /workspace -s /bin/bash ovh
|
|
|
1 |
+
# build with: docker build . --tag sctg/roco-idefics3:0.0.15 --tag sctg/roco-idefics3:latest --push
|
2 |
# run with
|
3 |
+
# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -e HF_TOKEN=hf_TOKEN -it sctg/roco-idefics3:0.0.15 bash -i /start.sh sleep infinity
|
4 |
+
# docker run --gpus all --user=42420:42420 -p 7000-8000:7000-8000 -p 8080:8080 -it sctg/roco-idefics3:0.0.15 bash -i /start.sh python /learn.py hf_...
|
5 |
FROM cloudflare/cloudflared:latest as cloudflared
|
6 |
+
FROM nvidia/cuda:12.6.2-devel-ubuntu22.04
|
7 |
ARG NODE_MAJOR="20"
|
8 |
RUN /usr/sbin/addgroup --gid 42420 ovh
|
9 |
RUN /usr/sbin/useradd -u 42420 --gid 42420 -m -d /workspace -s /bin/bash ovh
|
start.sh
CHANGED
@@ -27,12 +27,14 @@ fi
|
|
27 |
# Launch cloudflared tunnel
|
28 |
# CFTOKEN.enc is the Cloudflared encrypted token
|
29 |
# CFTOKEN.enc contained an encrypted string encrypted with the HF_TOKEN
|
|
|
|
|
30 |
if [ -f /workspace/.config/CFTOKEN.enc ]; then
|
31 |
echo "Decrypting CFTOKEN"
|
32 |
CFTOKEN_ENC=$(cat /workspace/.config/CFTOKEN.enc)
|
33 |
echo "$CFTOKEN_ENC" | openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" > /workspace/.config/cftoken
|
34 |
export CFTOKEN=$(cat /workspace/.config/cftoken)
|
35 |
-
/usr/local/bin/cloudflared --pidfile /tmp
|
36 |
fi
|
37 |
|
38 |
git clone https://huggingface.co/eltorio/IDEFICS3_ROCOv2
|
|
|
27 |
# Launch cloudflared tunnel
|
28 |
# CFTOKEN.enc is the Cloudflared encrypted token
|
29 |
# CFTOKEN.enc contained an encrypted string encrypted with the HF_TOKEN
|
30 |
+
# CFTOKEN.enc was created with:
|
31 |
+
# echo "the_cf_tunnel_token" | openssl aes-256-cbc -base64 -md sha256 -pass pass:"$HF_TOKEN" > CFTOKEN.enc
|
32 |
if [ -f /workspace/.config/CFTOKEN.enc ]; then
|
33 |
echo "Decrypting CFTOKEN"
|
34 |
CFTOKEN_ENC=$(cat /workspace/.config/CFTOKEN.enc)
|
35 |
echo "$CFTOKEN_ENC" | openssl aes-256-cbc -a -d -md sha256 -pass pass:"$HF_TOKEN" > /workspace/.config/cftoken
|
36 |
export CFTOKEN=$(cat /workspace/.config/cftoken)
|
37 |
+
/usr/local/bin/cloudflared --pidfile /tmp/cf.pid --autoupdate-freq 24h0m0s tunnel run --token $CFTOKEN &
|
38 |
fi
|
39 |
|
40 |
git clone https://huggingface.co/eltorio/IDEFICS3_ROCOv2
|