From f7dcf8fcea4d486544f221032625a97ad7dc5405 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Mon, 23 Mar 2020 13:58:49 -0400 Subject: [PATCH] [BertAbs] Move files around for more consistent naming --- examples/summarization/bertabs/configuration_bertabs.py | 2 +- .../bertabs/convert_bertabs_original_pytorch_checkpoint.py | 2 +- examples/summarization/bertabs/modeling_bertabs.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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", }