Skip to content

Commit 93f5636

Browse files
echarlaixyoung-developer
authored andcommitted
Fix test (huggingface#1772)
fix test
1 parent bc6a09d commit 93f5636

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/exporters/onnx/test_onnx_export.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -541,7 +541,7 @@ def test_custom_export_official_model(self):
541541

542542
@parameterized.expand([(None,), (fn_get_submodels_custom,)])
543543
def test_custom_export_trust_remote(self, fn_get_submodels):
544-
model_id = "fxmarty/tiny-mpt-random-remote-code"
544+
model_id = "echarlaix/tiny-mpt-random-remote-code"
545545
config = AutoConfig.from_pretrained(model_id, trust_remote_code=True)
546546
onnx_config = CustomMPTOnnxConfig(
547547
config=config,

0 commit comments

Comments
 (0)