wip deepseek r1

2025-03-02 22:01:24 +01:00
parent b7e95d7398
commit b64f8d7d33
40 changed files with 6654 additions and 1859 deletions
--- a/tools/train/train.sh
+++ b/tools/train/train.sh
@@ -0,0 +1,15 @@
+#!/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"
+
+train_deepseek --output-dir "$OUTPUT_DIR"