Skip to content

Commit 311f1b0

Browse files
committed
fix numpy version at 1.26.4 (with a pickle interoperability fix), because the result of unpickling will need to be read by vllm, which requires numpy<2.0.0
1 parent 5da0ea9 commit 311f1b0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dataset_openorca_mlperf_recipe/data_axs.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@
3333
]] ],
3434

3535
"python_deps": [
36+
[ "^^", "python_sync_pip_package", [[ "python_package", "package_name=numpy", "package_version===1.26.4" ]] ],
3637
[ "^^", "python_sync_pip_package", [[ "python_package", "package_name=datasets" ]] ],
3738
[ "^^", "python_sync_pip_package", [[ "python_package", "package_name=pandas" ]] ],
38-
[ "^^", "python_sync_pip_package", [[ "python_package", "package_name=transformers" ]] ],
39-
[ "^^", "python_sync_pip_package", [[ "python_package", "package_name=numpy" ]] ]
39+
[ "^^", "python_sync_pip_package", [[ "python_package", "package_name=transformers" ]] ]
4040
],
4141

4242
"newborn_entry_tags": [ "downloaded", "dataset_name=openorca" ],

0 commit comments

Comments
 (0)