From 5ad5518ca6416fe4734ff2ca7e9c0de2e0bb4bac Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Tue, 30 Jun 2026 18:30:23 -0400
Subject: [PATCH] fix: add deepspeed to auto-install dependencies

---
 train-on-this-server.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 0e3d40f..6f4c654 100644
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -30,7 +30,7 @@ echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..."
 pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
 
 echo "[3/4] Install training dependencies..."
-pip install transformers datasets trl peft accelerate bitsandbytes
+pip install transformers datasets trl peft accelerate bitsandbytes deepspeed
 
 # Check GPU
 echo ""