45 lines
1.2 KiB
Bash
45 lines
1.2 KiB
Bash
#!/usr/bin/env bash
|
||
# Pull additional AI models into Ollama
|
||
# Run after install: bash models/pull-models.sh --tier=entry
|
||
# ─────────────────────────────────────────────
|
||
TIER=${1:-entry}
|
||
|
||
echo "Pulling models for tier: $TIER"
|
||
|
||
entry_models=(
|
||
"llama3.1:8b" # General purpose, good baseline
|
||
"mistral:7b" # Fast, good for APIs
|
||
"llama3.1:70b" # Larger — only if enough VRAM (3× L40S has 144GB total)
|
||
"nomic-embed-text" # Embedding model for RAG
|
||
"codellama:13b" # Code generation
|
||
)
|
||
|
||
mid_models=(
|
||
"${entry_models[@]}"
|
||
"llama3.1:70b"
|
||
"mixtral:8x7b"
|
||
"deepseek-coder-v2:16b"
|
||
)
|
||
|
||
advanced_models=(
|
||
"${mid_models[@]}"
|
||
"llama3.1:405b"
|
||
"mixtral:8x22b"
|
||
)
|
||
|
||
case $TIER in
|
||
entry) models=("${entry_models[@]}") ;;
|
||
mid) models=("${mid_models[@]}") ;;
|
||
advanced) models=("${advanced_models[@]}") ;;
|
||
*) echo "Unknown tier: $TIER. Use entry, mid, or advanced."; exit 1 ;;
|
||
esac
|
||
|
||
for model in "${models[@]}"; do
|
||
echo ""
|
||
echo "→ Pulling $model..."
|
||
ollama pull "$model"
|
||
done
|
||
|
||
echo ""
|
||
echo "✓ All models pulled. List with: ollama list"
|