#!/bin/bash
set -e
cd /mnt/data/track_b_v5/corpus/gguf

HF=/mnt/data/track_b_ai_weights/venv/bin/hf

# Use the new hf CLI
download_gguf() {
  local key=$1
  local repo=$2
  local file=$3
  mkdir -p "$key"
  if [[ -f "$key/$file" ]] && [[ $(stat -c %s "$key/$file") -gt 1000000 ]]; then
    echo "  $key: already present ($(du -h "$key/$file" | cut -f1))"
    return
  fi
  echo "  downloading $key ($repo $file) ..."
  $HF download "$repo" "$file" --local-dir "$key" 2>&1 | tail -5
}

download_gguf "Qwen2.5-0.5B" "bartowski/Qwen2.5-0.5B-Instruct-GGUF" "Qwen2.5-0.5B-Instruct-Q4_K_M.gguf"
download_gguf "Qwen2.5-1.5B" "bartowski/Qwen2.5-1.5B-Instruct-GGUF" "Qwen2.5-1.5B-Instruct-Q4_K_M.gguf"
download_gguf "Qwen2.5-7B"   "bartowski/Qwen2.5-7B-Instruct-GGUF"   "Qwen2.5-7B-Instruct-Q4_K_M.gguf"
download_gguf "Llama-3.2-3B" "bartowski/Llama-3.2-3B-Instruct-GGUF" "Llama-3.2-3B-Instruct-Q4_K_M.gguf"
download_gguf "Mistral-7B"   "bartowski/Mistral-7B-Instruct-v0.3-GGUF" "Mistral-7B-Instruct-v0.3-Q4_K_M.gguf"

echo "GGUF totals:"
du -sh /mnt/data/track_b_v5/corpus/gguf/*/