Fix warning message in ElectraForCausalLM (#16023)

65f9653e · Pavel Belevich · GitHub · a69e1850 · 65f9653e
Unverified Commit 65f9653e authored Mar 09, 2022 by Pavel Belevich Committed by GitHub Mar 09, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/electra/modeling_electra.py src/transformers/models/electra/modeling_electra.py +1 -1

No files found.
--- a/src/transformers/models/electra/modeling_electra.py
+++ b/src/transformers/models/electra/modeling_electra.py
@@ -1517,7 +1517,7 @@ class ElectraForCausalLM(ElectraPreTrainedModel):
        super().__init__(config)

        if not config.is_decoder:
-            logger.warning("If you want to use `ElectraLMHeadModel` as a standalone, add `is_decoder=True.`")
+            logger.warning("If you want to use `ElectraForCausalLM` as a standalone, add `is_decoder=True.`")

        self.electra = ElectraModel(config)
        self.generator_predictions = ElectraGeneratorPredictions(config)