#!/bin/bash

# Volume Docker utilisé pour stocker tous les modèles
VOLUME="ollama_models"

# Liste des modèles à pré-télécharger
MODELS=(
  "codellama:13b"
  "wizardcoder:latest"
  "phi4:latest"
  "phi4-mini:latest"
  "thewindmom/hermes-3-llama-3.1-8b:latest"
  "llama3-chatqa:8b"
  "gemma3:12b-it-qat"
  "openchat:latest"
  "llava-llama3:latest"
  "llava:7b"
  "deepseek-coder-v2:16b"
  "gemma3:12b"
  "nomic-embed-text:latest"
)

for model in "${MODELS[@]}"; do
  echo "Téléchargement de $model ..."
  ollama pull "$model"
done

echo "Tous les modèles ont été téléchargés dans le volume $VOLUME."
