clean naming (#7068)

This commit is contained in:
Patrick von Platen
2020-09-11 09:57:53 +02:00
committed by GitHub
parent 8fcbe486e1
commit 221d4c63a3
6 changed files with 10 additions and 10 deletions

View File

@@ -99,7 +99,7 @@ CONFIG_MAPPING = OrderedDict(
AlbertConfig,
),
(
"bert-for-seq-generation",
"bert-generation",
BertGenerationConfig,
),
(

View File

@@ -12,7 +12,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
""" BertForSeqGeneration model configuration """
""" BertGeneration model configuration """
from .configuration_utils import PretrainedConfig
@@ -29,8 +29,8 @@ class BertGenerationConfig(PretrainedConfig):
Args:
vocab_size (:obj:`int`, `optional`, defaults to 50358):
Vocabulary size of the BertForSeqGeneration model. Defines the different tokens that
can be represented by the `inputs_ids` passed to the forward method of :class:`~transformers.BertForSeqGeneration`.
Vocabulary size of the BertGeneration model. Defines the different tokens that
can be represented by the `inputs_ids` passed to the forward method of :class:`~transformers.BertGeneration`.
hidden_size (:obj:`int`, `optional`, defaults to 1024):
Dimensionality of the encoder layers and the pooler layer.
num_hidden_layers (:obj:`int`, `optional`, defaults to 24):
@@ -60,7 +60,7 @@ class BertGenerationConfig(PretrainedConfig):
>>> from transformers import BertGenerationConfig, BertGenerationEncoder
>>> # Initializing a BertForSeqGeneration config
>>> # Initializing a BertGeneration config
>>> configuration = BertGenerationConfig()
>>> # Initializing a modelfrom the config
@@ -69,7 +69,7 @@ class BertGenerationConfig(PretrainedConfig):
>>> # Accessing the model configuration
>>> configuration = model.config
"""
model_type = "bert-for-seq-generation"
model_type = "bert-generation"
def __init__(
self,

View File

@@ -238,7 +238,7 @@ BERT_GENERATION_INPUTS_DOCSTRING = r"""
@add_start_docstrings(
"The bare BertForSeqGeneration model transformer outputting raw hidden-states without any specific head on top.",
"The bare BertGeneration model transformer outputting raw hidden-states without any specific head on top.",
BERT_GENERATION_START_DOCSTRING,
)
class BertGenerationEncoder(BertGenerationPreTrainedModel):

View File

@@ -12,7 +12,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
""" Tokenization class for model BertForSeqGeneration."""
""" Tokenization class for model BertGeneration."""
import os