Update run.zsh

run.zsh

@@ -1,43 +1,34 @@
 #!/bin/zsh
+set -e
 
-# Fast fine-tune for Zenith - World's First Autonomous AI Development Partner
+# Activate environment if using venv or conda
+# source venv/bin/activate
 
-PythonExe="python"
+echo "🚀 Starting Zenith fine-tuning..."
 
-echo "🚀 Setting up ZENITH fine-tune for Aspetos (<2h training)..."
-echo "Building the most advanced AI development partner in existence!"
-
-# Env vars for stable training (adjust if needed)
-export BASE_MODEL="/cloud/models/DeepSeek-Coder-V2-Lite-Instruct" # Please change this to the actual path of the model on your cloud GPU
+# Environment variables
+export BASE_MODEL="DeepSeek-Coder-V2-Lite-Instruct"
 export OUTPUT_DIR="outputs/zenith-lora"
 export DATA_PATH="data/zenith_combined.jsonl"
-export EPOCHS="1"
-export BATCH="4"  # Balanced for A100
-export GRAD_ACC="4"  # Effective batch size = 16
-export LR="1e-4"  # Stable learning rate for proper convergence
-export STEPS="200" # Increased steps for more training examples
-export MAX_SEQ_LEN="2048"
-export USE_4BIT="1"
-export SEED="42"
-export MAX_GRAD_NORM="1.0"  # Gradient clipping
-export WEIGHT_DECAY="0.01"
-export WARMUP_RATIO="0.05"
-export EARLY_STOP_PATIENCE="5"  # Allow more patience for learning
-export EVAL_STEPS="40"  # Balanced evaluation frequency
-export SAVE_STEPS="40"  # Balanced save frequency
+export VAL_PATH=""
+export STEPS=200
+export BATCH=2
+export GRAD_ACC=2
+export LR=1e-4
+export WARMUP_RATIO=0.05
+export WEIGHT_DECAY=0.01
+export MAX_GRAD_NORM=1.0
+export LOG_STEPS=10
+export SAVE_STEPS=50
+export SAVE_LIMIT=3
+export EVAL_STEPS=50
+export EARLY_STOP_PATIENCE=3
+export SEED=42
+export MAX_SEQ_LEN=2048
+export USE_4BIT=1
+export REPORT_TO="none"
 
-echo "Installing dependencies..."
-$PythonExe -m pip install -r requirements.txt
-if [ $? -ne 0 ]; then
-    echo "pip install failed"
-    exit 1
-fi
+# Run training
+python3 train.py | tee outputs/train_log.txt
 
-echo "🎯 Starting ZENITH fine-tune (target: 1800 steps, ~1-2 hours on GPU)..."
-echo "Training the world's most advanced autonomous development partner!"
-$PythonExe train.py
-if [ $? -ne 0 ]; then
-    echo "Training failed"
-    exit 1
-fi
-echo "✅ ZENITH training complete - Ready for deployment!"
+echo "✅ Training finished! Check logs at outputs/train_log.txt"