#!/bin/bash

set -e

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

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

mkdir -p "$OUTPUT_DIR"

/root/venvs/train/bin/python -m train.qwq --output-dir "$OUTPUT_DIR"