alverciito commited on
Commit
157adac
·
1 Parent(s): 4aa1cf7

fix huggingface model weight missmatch

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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
- self.post_init()
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()