diff --git a/tests/models/auto/test_modeling_auto.py b/tests/models/auto/test_modeling_auto.py index bc48a7b005ab..a788c1df98fc 100644 --- a/tests/models/auto/test_modeling_auto.py +++ b/tests/models/auto/test_modeling_auto.py @@ -579,9 +579,10 @@ def test_model_with_dotted_name_and_relative_imports(self): Test for issue #40496: AutoModel.from_pretrained() doesn't work for models with '.' in their name when there's a relative import. - Without the fix, this raises: ModuleNotFoundError: No module named 'transformers_modules.test-model_v1' + Without the fix, this raises: ModuleNotFoundError: + No module named 'transformers_modules.hf-internal-testing.remote_code_model_with_dots_v1' """ - model_id = "hf-internal-testing/remote_code_model_with_dots" + model_id = "hf-internal-testing/remote_code_model_with_dots_v1.0" model = AutoModel.from_pretrained(model_id, trust_remote_code=True) self.assertIsNotNone(model)