Unverified Commit 65f9653e authored by Pavel Belevich's avatar Pavel Belevich Committed by GitHub
Browse files

Fix warning message in ElectraForCausalLM (#16023)

parent a69e1850
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment