#!/usr/bin/env bash
# Ornith-1.5-122B lokal ausprobieren – macOS
# Automatisch erzeugt von Tokenstrom (tokenstrom.de).
# Aufruf: bash try_mac.sh            # Ollama (Standard)
#        MODE=hf bash try_mac.sh    # Python/transformers
set -e
cd "$(dirname "$0")"
echo "== Ornith-1.5-122B =="
echo "WARNUNG: Groß (125.1 Mrd. Parameter): braucht auch quantisiert ~76.6 GB VRAM/RAM. Colab Free (T4, 15 GB) reicht nicht."

if [ "${MODE:-ollama}" = "ollama" ]; then
  echo "== Ollama =="
  if ! command -v ollama >/dev/null 2>&1; then
    if command -v brew >/dev/null 2>&1; then brew install ollama; else
      echo "Bitte zuerst die Ollama-App installieren: https://ollama.com/download/mac"; exit 1; fi
  fi
  curl -s http://localhost:11434/api/tags >/dev/null 2>&1 || (ollama serve >/dev/null 2>&1 &)
  sleep 3
  MODEL="${MODEL:-}"
  # Passende Variante für deine Hardware auswählen (eigene Wahl:  MODEL=<name> bash try_mac.sh)
  if [ -z "${MODEL:-}" ]; then
  MEM_GB=$(( $(sysctl -n hw.memsize) / 1073741824 * 7 / 10 ))   # ca. 70 % des Speichers nutzt die GPU
  MODEL=$(printf '%s\n' "84.5 hf.co/ornith-ai/Ornith-1.5-122B-GGUF:Q4_K_M 1
99.0 hf.co/ornith-ai/Ornith-1.5-122B-GGUF:Q5_K_M 0
114.0 hf.co/ornith-ai/Ornith-1.5-122B-GGUF:Q6_K 0
147.0 hf.co/ornith-ai/Ornith-1.5-122B-GGUF:Q8_0 0
275.5 hf.co/ornith-ai/Ornith-1.5-122B-GGUF:BF16 0" | awk -v mem="$MEM_GB" '
    { need[NR]=$1+0; ref[NR]=$2; def[NR]=$3+0 }
    END { mem=mem+0; for (i=1;i<=NR;i++) if (def[i]==1 && need[i]<=mem) { print ref[i]; exit }
          best=ref[1]; for (i=1;i<=NR;i++) if (need[i]<=mem) best=ref[i]; print best }')
  echo "Verfügbarer Speicher: ca. ${MEM_GB} GB  ->  gewählte Variante: $MODEL"
  fi

  MODEL="${MODEL:-hf.co/ornith-ai/Ornith-1.5-122B-GGUF}"
  echo "Lade $MODEL herunter (beim ersten Mal dauert das je nach Größe einige Minuten) ..."
  ollama pull "$MODEL"
  echo ""
  echo "Fertig! Tippe jetzt deine Frage ein und drücke Enter. Beenden mit /bye"
  exec ollama run "$MODEL"
fi

echo "== Python / Hugging Face =="
python3 -m venv .venv-ornith-1-5-122b
source .venv-ornith-1-5-122b/bin/activate
pip install -q -U pip
pip install -q -U torch transformers accelerate

python3 demo.py