From b9faf2f93085e3cf2c65184a69d1d9e502f95786 Mon Sep 17 00:00:00 2001 From: dbleyl Date: Mon, 9 Jun 2025 11:23:20 -0400 Subject: [PATCH] Fix TypeError: 'NoneType' object is not iterable for esm (#38667) (#38668) Add post_init() calls to EsmForMaskedLM, EsmForTokenClassification and EsmForSequenceClassification. --- src/transformers/models/esm/modeling_esm.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/transformers/models/esm/modeling_esm.py b/src/transformers/models/esm/modeling_esm.py index 640b8429f00..7505268bb37 100755 --- a/src/transformers/models/esm/modeling_esm.py +++ b/src/transformers/models/esm/modeling_esm.py @@ -1023,6 +1023,8 @@ class EsmForMaskedLM(EsmPreTrainedModel): self.init_weights() + self.post_init() + def get_output_embeddings(self): return self.lm_head.decoder @@ -1127,6 +1129,8 @@ class EsmForSequenceClassification(EsmPreTrainedModel): self.init_weights() + self.post_init() + @auto_docstring def forward( self, @@ -1210,6 +1214,8 @@ class EsmForTokenClassification(EsmPreTrainedModel): self.init_weights() + self.post_init() + @auto_docstring def forward( self,