Update modeling_blaser.py
Browse files- modeling_blaser.py +3 -0
modeling_blaser.py
CHANGED
|
@@ -117,6 +117,9 @@ class BlaserModel(PreTrainedModel):
|
|
| 117 |
self.input_form = core.input_form
|
| 118 |
self.norm_emb = core.norm_emb
|
| 119 |
|
|
|
|
|
|
|
|
|
|
| 120 |
def forward(self, src, mt, ref=None):
|
| 121 |
# Use the same featurization as in BlaserCore
|
| 122 |
src = F.normalize(src) if self.norm_emb else src
|
|
|
|
| 117 |
self.input_form = core.input_form
|
| 118 |
self.norm_emb = core.norm_emb
|
| 119 |
|
| 120 |
+
# CRITICAL FIX: Ensures the model is fully registered for newer Transformers versions
|
| 121 |
+
self.post_init()
|
| 122 |
+
|
| 123 |
def forward(self, src, mt, ref=None):
|
| 124 |
# Use the same featurization as in BlaserCore
|
| 125 |
src = F.normalize(src) if self.norm_emb else src
|