init-container

Dockerfile

@@ -0,0 +1,28 @@
+#ARG  EXOSCALE_DOCKER_REGISTRY=registry.internal.exoscale.ch
+
+#FROM ${EXOSCALE_DOCKER_REGISTRY}/docker.io/library/ubuntu:latest
+
+FROM ubuntu:22.04
+
+# Install the necessary dependencies 
+RUN apt-get update && \
+    apt-get install -y \
+    bash \
+    python3 \
+    python3-pip \
+    curl \
+    unzip && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install AWS CLI
+RUN curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip" 
+RUN unzip awscliv2.zip && ./aws/install 
+
+# Copy the script which downloads the model from S3
+COPY download_model.sh /usr/local/bin/
+
+# Give execution permissions to the script
+RUN chmod +x /usr/local/bin/download_model.sh
+
+# Script to run 
+ENTRYPOINT [ "/usr/local/bin/download_model.sh" ]

download_model.sh

@@ -0,0 +1,17 @@
+#!/bin/bash
+
+set -e
+
+function download_object_from_bucket() {
+  local response
+
+  response=$(aws --debug s3 cp  \
+    "$LLM_BUCKET" \
+    "$LOCAL_PATH" \
+    --recursive \
+    --endpoint-url $AWS_ENDPOINT_URL \
+    --region $AWS_DEFAULT_REGION)
+
+}
+
+download_object_from_bucket