[chore] update torch version to 2.9 (#12969)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
This commit is contained in:
Fan Yin
2025-11-26 06:47:34 +08:00
committed by GitHub
parent 64a11303ce
commit 36b1bcd242
13 changed files with 44 additions and 87 deletions

View File

@@ -62,10 +62,10 @@ dependencies = [
"tiktoken",
"timm==1.0.16",
"torch_memory_saver==0.0.9",
"torch==2.8.0",
"torch==2.9.1",
"torchcodec==0.7.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.
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' and platform_machine == 'armv7l')",
"torchaudio==2.8.0",
"torchaudio==2.9.1",
"torchvision",
"torchao==0.9.0",
"tqdm",