diff --git a/requirements-cpu.txt b/requirements-cpu.txt index db8ad9d3a015..e62f31329776 100644 --- a/requirements-cpu.txt +++ b/requirements-cpu.txt @@ -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 \ No newline at end of file +torchvision; platform_machine != "ppc64le" # required for the image processor of phi3v, this must be updated alongside torch +datasets # for benchmark scripts \ No newline at end of file