From 1786830d84962e94a8d1645c3e58527231bee1ea Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Tue, 30 Jun 2026 16:00:17 -0400
Subject: [PATCH] feat: unload AI model before training

---
 train-on-this-server.sh | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 79ad3c6..10154db 100644
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -10,6 +10,11 @@ echo "=== LoRA Training Setup ==="
 echo "Repo: ${REPO_DIR}"
 echo ""
 
+# Unload AI model to free VRAM
+echo "=== Unloading AI model ==="
+ai none
+sleep 2
+
 # Check Python
 if ! command -v python3 &> /dev/null; then
     echo "ERROR: python3 not found"