Changed "mixtral_moe.onn" to model_path

microsoft · Jun 28, 2024 · 3183dde · 3183dde
1 parent 2de42ed
commit 3183dde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/test/python/transformers/test_parity_mixtral_moe.py b/onnxruntime/test/python/transformers/test_parity_mixtral_moe.py
@@ -43,7 +43,7 @@ def print_tensor(name, numpy_array):
 
 
 def save_model_to_disk(model, model_path):
-    external_data_path = "mixtral_moe.onnx" + ".data"
+    external_data_path = model_path + ".data"
     onnx.save_model(
         model, model_path, save_as_external_data=True, all_tensors_to_one_file=True, location=external_data_path
     )