GPU pipelines minor fix (#19)

Author: ultmaster

examples/calc_x/calc_agent_dev.py

@@ -28,4 +28,4 @@ def dev_task_loader() -> DevTaskLoader:
 
 
 if __name__ == "__main__":
-    Trainer(n_workers=2, dev=True, max_tasks=2).fit(CalcAgent(), "http://localhost:9999/", dev_task_loader())
+    Trainer(n_workers=1, dev=True, max_tasks=2).fit(CalcAgent(), "http://localhost:9999/", dev_task_loader())

scripts/restart_ray.sh

@@ -1,4 +1,7 @@
 #!/bin/bash
 
-ray stop --force
+set -ex
+
+ray stop -v --force --grace-period 60
+ps aux
 env RAY_DEBUG=legacy HYDRA_FULL_ERROR=1 VLLM_USE_V1=1 ray start --head --dashboard-host=0.0.0.0

scripts/setup_latest_gpu.sh

@@ -4,7 +4,7 @@ python -m pip install --upgrade --no-cache-dir pip
 
 pip install --no-cache-dir packaging ninja numpy pandas ipython ipykernel gdown wheel setuptools
 # This has to be pinned for VLLM to work.
-pip install --no-cache-dir torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128
+pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
 pip install --no-cache-dir flash-attn --no-build-isolation
 pip install --no-cache-dir vllm
 

scripts/setup_stable_gpu.sh

@@ -4,6 +4,7 @@ python -m pip install --upgrade pip
 
 pip install --no-cache-dir packaging ninja numpy pandas ipython ipykernel gdown wheel setuptools
 pip install --no-cache-dir torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128
+pip install --no-cache-dir transformers==4.53.3
 pip install --no-cache-dir flash-attn==2.8.1 --no-build-isolation
 pip install --no-cache-dir vllm==0.9.2