[VERL Use Case] Add torch_memory_saver into deps (#6247)

2025-05-12 19:09:03 -07:00
parent 3c32895cbe
commit 1ab14c4c5c
2 changed files with 1 additions and 2 deletions
--- a/scripts/ci_install_dependency.sh
+++ b/scripts/ci_install_dependency.sh
@@ -19,7 +19,6 @@ rm -rf /usr/local/lib/python3.10/dist-packages/sgl_kernel*
 pip install -e "python[all]"

 # Install additional dependencies
-pip install torch_memory_saver
 pip install transformers==4.51.0 timm torchaudio==2.6.0 sentence_transformers accelerate peft pandas datasets mooncake-transfer-engine==0.3.0

 # For compiling xgrammar kernels