diff --git a/python/pyproject.toml b/python/pyproject.toml
index 804b4291c..57954507e 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -60,7 +60,7 @@ dependencies = [
   "tiktoken",
   "timm==1.0.16",
   "torch==2.8.0",
-  "torch_memory_saver==0.0.9rc2",
+  "torch_memory_saver==0.0.9",
   "torchao==0.9.0",
   "torchaudio==2.8.0",
   "torchvision",