[Feature] Option to save model weights to CPU when memory saver mode is enabled (#10873)

Co-authored-by: molocule <34072934+molocule@users.noreply.github.com>
2025-10-03 04:48:19 -04:00
parent 34151f173b
commit 8c57490210
7 changed files with 78 additions and 22 deletions
--- 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.8",
+  "torch_memory_saver==0.0.9rc1",
  "torchao==0.9.0",
  "torchaudio==2.8.0",
  "torchvision",