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"