Fix Copied from

mfuntowicz · mfuntowicz · commit 49916c9019f6 · 2021-06-28T20:08:09.000+02:00
diff --git a/src/transformers/models/albert/configuration_albert.py b/src/transformers/models/albert/configuration_albert.py
@@ -155,7 +155,7 @@ def __init__(
         self.position_embedding_type = position_embedding_type
 
 
-# Copied from transformers.models.bert.configuration_bert.BertOnnxConfig with Bert->Albert
+# Copied from transformers.models.bert.configuration_bert.BertOnnxConfig with Roberta->Albert
 class AlbertOnnxConfig(OnnxConfig):
     @property
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
diff --git a/src/transformers/models/distilbert/configuration_distilbert.py b/src/transformers/models/distilbert/configuration_distilbert.py
@@ -139,6 +139,7 @@ def num_hidden_layers(self):
         return self.n_layers
 
 
+# Copied from transformers.models.bert.configuration_bert.BertOnnxConfig with Roberta->DistilBert
 class DistilBertOnnxConfig(OnnxConfig):
     @property
     def inputs(self) -> Mapping[str, Mapping[int, str]]:
