diff --git a/examples/summarization/bertabs/configuration_bertabs.py b/examples/summarization/bertabs/configuration_bertabs.py index c976180b2f..77acc84e8f 100644 --- a/examples/summarization/bertabs/configuration_bertabs.py +++ b/examples/summarization/bertabs/configuration_bertabs.py @@ -23,7 +23,7 @@ logger = logging.getLogger(__name__) BERTABS_FINETUNED_CONFIG_MAP = { - "bertabs-finetuned-cnndm": "https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization-config.json", + "bertabs-finetuned-cnndm": "https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization/config.json", } diff --git a/examples/summarization/bertabs/convert_bertabs_original_pytorch_checkpoint.py b/examples/summarization/bertabs/convert_bertabs_original_pytorch_checkpoint.py index a1cbd64dd8..d94f6f0176 100644 --- a/examples/summarization/bertabs/convert_bertabs_original_pytorch_checkpoint.py +++ b/examples/summarization/bertabs/convert_bertabs_original_pytorch_checkpoint.py @@ -157,7 +157,7 @@ def convert_bertabs_checkpoints(path_to_checkpoints, dump_path): # directory structure. We save the state_dict instead. logging.info("saving the model's state dictionary") torch.save( - new_model.state_dict(), "bertabs-finetuned-cnndm-extractive-abstractive-summarization-pytorch_model.bin" + new_model.state_dict(), "./bertabs-finetuned-cnndm-extractive-abstractive-summarization/pytorch_model.bin" ) diff --git a/examples/summarization/bertabs/modeling_bertabs.py b/examples/summarization/bertabs/modeling_bertabs.py index 117180d69b..d89604be56 100644 --- a/examples/summarization/bertabs/modeling_bertabs.py +++ b/examples/summarization/bertabs/modeling_bertabs.py @@ -34,7 +34,7 @@ from transformers import BertConfig, BertModel, PreTrainedModel MAX_SIZE = 5000 BERTABS_FINETUNED_MODEL_MAP = { - "bertabs-finetuned-cnndm": "https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization-pytorch_model.bin", + "bertabs-finetuned-cnndm": "https://s3.amazonaws.com/models.huggingface.co/bert/remi/bertabs-finetuned-cnndm-extractive-abstractive-summarization/pytorch_model.bin", }