requirements: use stable torch packages on s390x (#26864)

2026-08-11 21:58:53 +08:00
parent 5d16e81dd9
commit 2468576f24
4 changed files with 3 additions and 15 deletions
@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
 torch
-torchvision
+torchvision; platform_machine != "s390x"
 transformers
 huggingface-hub
 accelerate