diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index fd9b0b1..5f8a970 100755
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -78,19 +78,8 @@ echo "Press Ctrl+C to stop (model will be saved at checkpoint)."
 echo ""
 
 # Use torchrun for distributed training (2 GPUs)
-# Add --check-only to validate without training:
-# torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml --check-only
-
-# Run in screen if available (bypasses cgroup limits)
-if command -v screen &> /dev/null; then
-    echo "Running in screen session..."
-    screen -dmS lora_training bash -c "torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml"
-    echo "Training started in screen session 'lora_training'"
-    echo "Watch with: screen -r lora_training"
-else
-    echo "Screen not available, running directly..."
-    torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml
-fi
+echo "Starting training..."
+torchrun --nproc_per_node=2 training/scripts/train.py --config training/configs/ornith-35b-lora.yaml
 
 echo ""
 echo "=== Training Complete ==="