torchao: update to v0.7.0

bghira · Dec 18, 2024 · 3662525 · 3662525
1 parent 020bf37
commit 3662525
Show file tree

Hide file tree

Showing 6 changed files with 37 additions and 27 deletions.
diff --git a/install/apple/poetry.lock b/install/apple/poetry.lock
diff --git a/install/apple/pyproject.toml b/install/apple/pyproject.toml
@@ -43,7 +43,7 @@ lycoris-lora = {git = "https://github.com/kohakublueleaf/lycoris", rev = "dev"}
 fastapi = {extras = ["standard"], version = "^0.115.0"}
 deepspeed = "^0.15.1"
 sentencepiece = "^0.2.0"
-torchao = "^0.5.0"
+torchao = "^0.7.0"
 torchaudio = "^2.5.0"
 atomicwrites = "^1.4.1"
 

diff --git a/install/rocm/poetry.lock b/install/rocm/poetry.lock
diff --git a/install/rocm/pyproject.toml b/install/rocm/pyproject.toml
@@ -45,6 +45,7 @@ torch-optimi = "^0.2.1"
 fastapi = {extras = ["standard"], version = "^0.115.0"}
 bitsandbytes = "^0.44.1"
 atomicwrites = "^1.4.1"
+torchao = "^0.7.0"
 
 
 [build-system]

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -46,7 +46,7 @@ lycoris-lora = {git = "https://github.com/kohakublueleaf/lycoris", rev = "dev"}
 torch-optimi = "^0.2.1"
 toml = "^0.10.2"
 fastapi = {extras = ["standard"], version = "^0.115.0"}
-torchao = {version = "^0.5.0+cu124", source = "pytorch"}
+torchao = "^0.7.0"
 lm-eval = "^0.4.4"
 nvidia-cudnn-cu12 = "*"
 nvidia-nccl-cu12 = "*"