[hotfix] Fixes on cuda 13 docker image (#17541)

Co-authored-by: iforgetmyname <iforgetmyname@users.noreply.github>
2026-01-22 12:29:55 +08:00
parent a5bbcda968
commit fafa171529
5 changed files with 5 additions and 0 deletions
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -66,6 +66,7 @@ dependencies = [
  "timm==1.0.16",
  "torch_memory_saver==0.0.9",
  "torch==2.9.1",
+  "torchao==0.9.0",
  "torchaudio==2.9.1",
  "torchcodec==0.8.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torchcodec does not exist in those systems. If not provided, transformer will use torchvision instead by default.
  "torchvision",
--- a/python/pyproject_cpu.toml
+++ b/python/pyproject_cpu.toml
@@ -57,6 +57,7 @@ dependencies = [
  "soundfile==0.13.1",
  "tiktoken",
  "timm==1.0.16",
+  "torchao==0.9.0",
  "tqdm",
  "transformers==4.57.1",
  "uvicorn",
--- a/python/pyproject_other.toml
+++ b/python/pyproject_other.toml
@@ -57,6 +57,7 @@ runtime_common = [
  "soundfile==0.13.1",
  "tiktoken",
  "timm==1.0.16",
+  "torchao==0.9.0",
  "tqdm",
  "transformers==4.57.1",
  "uvicorn",
--- a/python/pyproject_xpu.toml
+++ b/python/pyproject_xpu.toml
@@ -61,6 +61,7 @@ dependencies = [
  "soundfile==0.13.1",
  "tiktoken",
  "timm==1.0.16",
+  "torchao==0.9.0",
  "tqdm",
  "transformers==4.57.1",
  "uvicorn",