Fix test (#1772)

fix test
huggingface · Mar 25, 2024 · 8896bb1 · 8896bb1
1 parent 8a95414
commit 8896bb1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,