update pytorch-xpu to 2.12 (#27133)
Co-authored-by: Ma Mingfei <mingfei.ma@intel.com> Co-authored-by: MingxuZh <109504044+MingxuZh@users.noreply.github.com>
This commit is contained in:
co-authored by
Ma Mingfei
MingxuZh
parent
77c4d53f19
commit
bcd9c5a903
@@ -15,11 +15,11 @@ classifiers = [
|
||||
]
|
||||
|
||||
dependencies = [
|
||||
"torch==2.11.0+xpu",
|
||||
"torchcodec==0.11.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. torch==2.11.0 on XPU uses 0.11.0
|
||||
"torch==2.12.0+xpu",
|
||||
"torchcodec==0.12.0 ; sys_platform != 'linux' or (sys_platform == 'linux' and platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')", # torch==2.12.0 on XPU uses torchcodec 0.12.0
|
||||
"av ; sys_platform == 'linux' and (platform_machine == 'aarch64' or platform_machine == 'arm64' or platform_machine == 'armv7l')",
|
||||
"torchaudio==2.11.0+xpu",
|
||||
"torchvision",
|
||||
"torchvision==0.27.0+xpu",
|
||||
"sgl-kernel @ git+https://github.com/sgl-project/sgl-kernel-xpu.git",
|
||||
"IPython",
|
||||
"aiohttp",
|
||||
@@ -60,7 +60,7 @@ dependencies = [
|
||||
"soundfile==0.13.1",
|
||||
"tiktoken",
|
||||
"timm==1.0.16",
|
||||
"torchao==0.9.0+xpu",
|
||||
"torchao==0.17.0+xpu",
|
||||
"tqdm",
|
||||
"mistral_common>=1.11.0",
|
||||
"transformers==5.8.1",
|
||||
|
||||
Reference in New Issue
Block a user