riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-04 05:10:06 +06:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Fix TypeError: 'NoneType' object is not iterable for esm (#38667) (#38668)

Browse Source 
Add post_init() calls to EsmForMaskedLM, EsmForTokenClassification and EsmForSequenceClassification.
This commit is contained in:
dbleyl 2025-06-09 11:23:20 -04:00 committed by GitHub
parent 11dca07a10
commit b9faf2f930
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/models/esm/modeling_esm.py
View File

@ -1023,6 +1023,8 @@ class EsmForMaskedLM(EsmPreTrainedModel):
self.init_weights() self.init_weights()
self.post_init()
def get_output_embeddings(self): def get_output_embeddings(self):
return self.lm_head.decoder return self.lm_head.decoder
@ -1127,6 +1129,8 @@ class EsmForSequenceClassification(EsmPreTrainedModel):
self.init_weights() self.init_weights()
self.post_init()
@auto_docstring @auto_docstring
def forward( def forward(
self, self,
@ -1210,6 +1214,8 @@ class EsmForTokenClassification(EsmPreTrainedModel):
self.init_weights() self.init_weights()
self.post_init()
@auto_docstring @auto_docstring
def forward( def forward(
self, self,