Skip to content

GitLab

Unverified Commit a7043c6f authored by fzyzcjy's avatar fzyzcjy Committed by GitHub
Browse files

Bump torch_memory_saver to avoid installing pre-release versions (#11797)

parent 67e34c56
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/pyproject.toml
View file @ a7043c6f
...@@ -60,7 +60,7 @@ dependencies = [ ...@@ -60,7 +60,7 @@ dependencies = [
"tiktoken", "tiktoken",
"timm==1.0.16", "timm==1.0.16",
"torch==2.8.0", "torch==2.8.0",
"torch_memory_saver==0.0.9rc2", "torch_memory_saver==0.0.9",
"torchao==0.9.0", "torchao==0.9.0",
"torchaudio==2.8.0", "torchaudio==2.8.0",
"torchvision", "torchvision",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment