Fixup post init (for v5 remot compatibility)
Browse filesAs per title, in v5 `post_init` became mandatory for everything inheriting from pretrained model
- 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(
|