From 9c4f0593b74eedb88ac93c8e3c0f0a62e24730a0 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Mon, 25 Mar 2024 11:56:49 +0100 Subject: [PATCH] fix test --- tests/exporters/onnx/test_onnx_export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/exporters/onnx/test_onnx_export.py b/tests/exporters/onnx/test_onnx_export.py index b2b6cbba28..9eddc4c86d 100644 --- a/tests/exporters/onnx/test_onnx_export.py +++ b/tests/exporters/onnx/test_onnx_export.py @@ -541,7 +541,7 @@ def test_custom_export_official_model(self): @parameterized.expand([(None,), (fn_get_submodels_custom,)]) def test_custom_export_trust_remote(self, fn_get_submodels): - model_id = "fxmarty/tiny-mpt-random-remote-code" + model_id = "echarlaix/tiny-mpt-random-remote-code" config = AutoConfig.from_pretrained(model_id, trust_remote_code=True) onnx_config = CustomMPTOnnxConfig( config=config,