#!/bin/bash

set -e

SIA_DIR="/root/sia"
OUTPUT_DIR="${1:-/root/models/$(cd "$SIA_DIR" && git rev-parse HEAD)}"

echo "Output dir: $OUTPUT_DIR"

if [ -n "$(cd "$SIA_DIR" && git status --porcelain)" ]; then
    echo "Uncommitted changes in SIA directory"
    #exit 1
fi

mkdir -p "$OUTPUT_DIR"

gemma_train --model "google/gemma-3-12b-it" --output-dir "$OUTPUT_DIR"
cp "$OUTPUT_DIR"/tokenizer/* "$OUTPUT_DIR"/merged/
llama-convert-hf-to-gguf --outfile "$OUTPUT_DIR/model.gguf" --outtype q8_0 "$OUTPUT_DIR/merged"