From 1095032801a65b11c3f1cfc3c4a9e632d85dedba Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Wed, 1 Jul 2026 16:24:30 -0400
Subject: [PATCH] feat: add ai none to unload AI model before training

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

diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index bd0439c..6dd219e 100755
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -3,6 +3,10 @@
 
 set -e
 
+# Unload AI model to free GPU memory
+echo "Unloading AI model..."
+ai none
+
 echo "=== Cyron LoRA Training Setup ==="
 
 # Create venv