Let more then one model beeing pushed

embed_model/Dockerfile

@@ -1,10 +1,11 @@
 FROM svenbrnn/runpod-ollama:0.5.7
 
-ARG MODEL_NAME
-ENV MODEL_NAME=$MODEL_NAME
+ARG MODEL_NAMES
+ENV MODEL_NAMES=$MODEL_NAMES
 ADD preload_model.sh /preload_model.sh
+RUN apt-get update && apt-get install bash
 RUN chmod +x /preload_model.sh && /preload_model.sh
 
 # Copy the model to the volume
 FROM svenbrnn/runpod-ollama:0.5.7
-COPY --from=0 /runpod-volume /runpod-volume
+COPY --from=0 /runpod-volume /runpod-volume

embed_model/README.md

@@ -1,3 +1,3 @@
 # Embed model into runpod-worker-ollama
 
-Run ``docker build --build-arg MODEL_NAME="<model-name>" -t <yourname>/<yourrepo>:<tag> .`` 
+Run ``docker build --build-arg MODEL_NAMES="<model-name>,..." -t <yourname>/<yourrepo>:<tag> .`` 

embed_model/preload_model.sh

@@ -1,7 +1,7 @@
-#!/bin/sh
+#!/bin/bash
 
 # Start the Ollama server in the background
-echo "Starting Ollama server to preload model: $MODEL_NAME"
+echo "Starting Ollama server to preload models: $MODEL_NAMES"
 ollama serve &
 
 # Capture the PID of the Ollama server
@@ -11,15 +11,20 @@ OLLAMA_PID=$!
 echo "Waiting for Ollama server to start..."
 sleep 5
 
-# Pull the specified model
-echo "Pulling model: $MODEL_NAME"
-if ollama pull "$MODEL_NAME"; then
-  echo "Successfully pulled model: $MODEL_NAME"
-else
-  echo "Failed to pull model: $MODEL_NAME"
-  kill $OLLAMA_PID
-  exit 1
-fi
+# Split the comma-separated model names into an array
+IFS=',' read -r -a MODELS <<< "$MODEL_NAMES"
+
+# Loop through each model and pull it
+for MODEL_NAME in "${MODELS[@]}"; do
+  echo "Pulling model: $MODEL_NAME"
+  if ollama pull "$MODEL_NAME"; then
+    echo "Successfully pulled model: $MODEL_NAME"
+  else
+    echo "Failed to pull model: $MODEL_NAME"
+    kill $OLLAMA_PID
+    exit 1
+  fi
+done
 
 # Stop the Ollama server
 echo "Stopping Ollama server..."