AntonV HF Staff commited on
Commit
61f9bb9
·
verified ·
1 Parent(s): cd915ad

Fixup post init (for v5 remot compatibility)

Browse files

As per title, in v5 `post_init` became mandatory for everything inheriting from pretrained model

Files changed (1) hide show
  1. modeling_xlm_roberta.py +1 -0
modeling_xlm_roberta.py CHANGED
@@ -480,6 +480,7 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
480
  self.name_or_path, trust_remote_code=True
481
  )
482
  self._rotary_emb_base = config.rotary_emb_base
 
483
 
484
  @torch.inference_mode()
485
  def encode(
 
480
  self.name_or_path, trust_remote_code=True
481
  )
482
  self._rotary_emb_base = config.rotary_emb_base
483
+ self.post_init()
484
 
485
  @torch.inference_mode()
486
  def encode(