diff --git a/docs/source/en/model_doc/marian.md b/docs/source/en/model_doc/marian.md index 8078ea1427c9..d8ebec8ffb0a 100644 --- a/docs/source/en/model_doc/marian.md +++ b/docs/source/en/model_doc/marian.md @@ -105,7 +105,7 @@ from huggingface_hub import list_models model_list = list_models() org = "Helsinki-NLP" -model_ids = [x.modelId for x in model_list if x.modelId.startswith(org)] +model_ids = [x.id for x in model_list if x.id.startswith(org)] suffix = [x.split("/")[1] for x in model_ids] old_style_multi_models = [f"{org}/{s}" for s in suffix if s != s.lower()] ``` diff --git a/src/transformers/models/marian/convert_marian_to_pytorch.py b/src/transformers/models/marian/convert_marian_to_pytorch.py index 593162ffe674..f086e480dfff 100644 --- a/src/transformers/models/marian/convert_marian_to_pytorch.py +++ b/src/transformers/models/marian/convert_marian_to_pytorch.py @@ -65,7 +65,7 @@ def find_pretrained_model(src_lang: str, tgt_lang: str) -> List[str]: """Find models that can accept src_lang as input and return tgt_lang as output.""" prefix = "Helsinki-NLP/opus-mt-" model_list = list_models() - model_ids = [x.modelId for x in model_list if x.modelId.startswith("Helsinki-NLP")] + model_ids = [x.id for x in model_list if x.id.startswith("Helsinki-NLP")] src_and_targ = [ remove_prefix(m, prefix).lower().split("-") for m in model_ids if "+" not in m ] # + cant be loaded. diff --git a/tests/models/marian/test_modeling_marian.py b/tests/models/marian/test_modeling_marian.py index a2668e663b6b..4f4fa36622d1 100644 --- a/tests/models/marian/test_modeling_marian.py +++ b/tests/models/marian/test_modeling_marian.py @@ -409,7 +409,7 @@ class ModelManagementTests(unittest.TestCase): @require_torch def test_model_names(self): model_list = list_models() - model_ids = [x.modelId for x in model_list if x.modelId.startswith(ORG_NAME)] + model_ids = [x.id for x in model_list if x.id.startswith(ORG_NAME)] bad_model_ids = [mid for mid in model_ids if "+" in model_ids] self.assertListEqual([], bad_model_ids) self.assertGreater(len(model_ids), 500) diff --git a/utils/update_tiny_models.py b/utils/update_tiny_models.py index 2cd8f72d56ae..9dc4cdf6e6b2 100644 --- a/utils/update_tiny_models.py +++ b/utils/update_tiny_models.py @@ -94,7 +94,7 @@ def get_tiny_model_summary_from_hub(output_path): ) _models = set() for x in models: - model = x.modelId + model = x.id org, model = model.split("/") if not model.startswith("tiny-random-"): continue