diff --git a/python/pyproject.toml b/python/pyproject.toml
index fb095f2a0..e74ae7625 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -58,7 +58,7 @@ dependencies = [
   "tiktoken",
   "timm==1.0.16",
   "torch==2.8.0",
-  "torch_memory_saver==0.0.9rc1",
+  "torch_memory_saver==0.0.9rc2",
   "torchao==0.9.0",
   "torchaudio==2.8.0",
   "torchvision",