Files
pmomusic/LLM/Docker/start.sh
2025-08-17 21:24:53 +02:00

40 lines
1.0 KiB
Bash

#!/bin/bash
set -e
# Lancer Ollama en arrière-plan
export OLLAMA_MODELS=/models
echo "🔹 Démarrage de Ollama..." 1>&2
ollama serve | sed 's/^/ 🔹[Ollama server] /' &
sleep 10
echo "🔹 Preaload Ollama models: "
ollama ls | sed 's/^/ 🔹 /' 1>&2
# Attendre Ollama
sleep 5
# Vérifier / précharger le modèle Nomic Embed Text
EMBED_MODEL="nomic-embed-text:latest"
echo "🔹 Vérification du modèle d'embedding: $EMBED_MODEL" 1>&2
if ! ollama list | grep -q "$EMBED_MODEL"; then
echo " 🔹 Modèle $EMBED_MODEL non trouvé, téléchargement..." 1>&2
ollama pull "$EMBED_MODEL"
else
echo " 🔹 Modèle $EMBED_MODEL déjà présent" 1>&2
fi
echo "🔹 Vérification du modèle LLM: $OLLAMA_MODEL" 1>&2
if ! ollama list | grep -q "$OLLAMA_MODEL"; then
echo " 🔹 Modèle $OLLAMA_MODEL non trouvé, téléchargement..." 1>&2
ollama pull "$OLLAMA_MODEL"
else
echo " 🔹 Modèle $OLLAMA_MODEL déjà présent" 1>&2
fi
# Lancer FastAPI
echo "🔹 Démarrage de FastAPI..." 1>&2
exec uvicorn app:app --host 0.0.0.0 --port 8000 --reload