Files
SIA/scripts/train.sh
2025-05-22 18:22:15 +02:00

19 lines
511 B
Bash
Executable File

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