Skip to content

Commit

Permalink
[Bugfix][CI][CPU] add missing datasets package to requirements-cpu.txt (
Browse files Browse the repository at this point in the history
#11159)

Signed-off-by: jiang1.li <[email protected]>
  • Loading branch information
bigPYJ1151 authored Dec 13, 2024
1 parent 7cd7409 commit f93bf2b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion requirements-cpu.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,5 @@
# Dependencies for CPUs
torch==2.5.1+cpu; platform_machine != "ppc64le" and platform_machine != "aarch64"
torch==2.5.1; platform_machine == "aarch64"
torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch
torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch
datasets # for benchmark scripts

0 comments on commit f93bf2b

Please sign in to comment.