diff --git a/onnxruntime/test/python/transformers/test_optimizer_utils.py b/onnxruntime/test/python/transformers/test_optimizer_utils.py index 70d26cf7b4af3..17363829eab1a 100644 --- a/onnxruntime/test/python/transformers/test_optimizer_utils.py +++ b/onnxruntime/test/python/transformers/test_optimizer_utils.py @@ -13,7 +13,7 @@ class TestOptimizerUtils(unittest.TestCase): def test_extract_raw_data_from_model(self): - model = self._get_model_proto_with_raw_data() + model = self._get_model_proto_with_raw_data(False) external_names, external_values = extract_raw_data_from_model(model) self.assertEqual(list(external_names), ["inputs"]) self.assertEqual(len(external_values), 1)