2024-04-05 15:39:17 +08:00
|
|
|
# Common dependencies
|
|
|
|
|
-r requirements-common.txt
|
|
|
|
|
|
|
|
|
|
# Dependencies for x86_64 CPUs
|
2024-06-27 05:53:04 +08:00
|
|
|
torch == 2.3.1+cpu; platform_machine != "ppc64le"
|
|
|
|
|
torchvision == 0.18.1+cpu; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch
|
|
|
|
|
triton >= 2.2.0 # FIXME(woosuk): This is a hack to avoid import error.
|