alverciito commited on
Commit
33c844e
·
1 Parent(s): 157adac

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()