alverciito
commited on
Commit
·
33c844e
1
Parent(s):
157adac
fix huggingface model weight missmatch
Browse files
model.py
CHANGED
|
@@ -143,7 +143,7 @@ class SentenceCoseNet(PreTrainedModel):
|
|
| 143 |
self.model = SegmentationNetwork(self.to_model_config(config))
|
| 144 |
|
| 145 |
# Initialize weights following HF conventions
|
| 146 |
-
|
| 147 |
|
| 148 |
# Set evaluation mode by default
|
| 149 |
self.model.eval()
|
|
|
|
| 143 |
self.model = SegmentationNetwork(self.to_model_config(config))
|
| 144 |
|
| 145 |
# Initialize weights following HF conventions
|
| 146 |
+
self.post_init()
|
| 147 |
|
| 148 |
# Set evaluation mode by default
|
| 149 |
self.model.eval()
|