#!/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"

train_gemma --output-dir "$OUTPUT_DIR"