Update numpy version to 2.0.2, fix --dataset-file path.

krai · Sep 6, 2024 · 0f0c1d4 · 0f0c1d4
1 parent 0af71bd
commit 0f0c1d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/base_mixtral_loadgen_experiment/data_axs.json b/base_mixtral_loadgen_experiment/data_axs.json
@@ -39,7 +39,7 @@
 	[ "byname", "python_script" ],
 	[ "run", [], {
 	    "python_deps": [
-		[ "AS^IS", "^^", "python_sync_pip_package", "python_package,package_name=numpy,package_version=1.24.1" ],
+		[ "AS^IS", "^^", "python_sync_pip_package", "python_package,package_name=numpy,package_version=2.0.2" ],
 		[ "AS^IS", "^^", "python_sync_pip_package", "python_package,package_name=transformers,package_version=4.41.2" ],
 		[ "AS^IS", "^^", "python_sync_pip_package", "python_package,package_name=evaluate,package_version=0.4.0" ],
 		[ "AS^IS", "^^", "python_sync_pip_package", "python_package,package_name=accelerate,package_version=0.21.0"],
@@ -57,7 +57,7 @@
 	    ],
             "desired_python_version": "3.9",
             "abs_script_path": ["^^", "get", "abs_script_path"],
-            "script_extra_params": [ "^^", "substitute", "--mlperf-accuracy-file \"#{accuracy_log_path}#\" --checkpoint-path \"#{checkpoint_path}#\" --dataset-file \"#{dataset_pkl_path}#\" --dtype \"#{dtype_value}#\" #{n_workers_cmd}#" ],
+            "script_extra_params": [ "^^", "substitute", "--mlperf-accuracy-file \"#{accuracy_log_path}#\" --checkpoint-path \"#{checkpoint_path}#\" --dataset-file \"#{dataset_pkl_path}#/#{dataset_pkl_file_name}#\" --dtype \"#{dtype_value}#\" #{n_workers_cmd}#" ],
 	    "capture_output": true
 	} ],
 	0,