bugfix ModelForCrossEncoding

leftshiftone · Jul 23, 2023 · 3fa2531 · 3fa2531
1 parent 108166e
commit 3fa2531
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnx_transformers/model_for_cross_encoding.py b/onnx_transformers/model_for_cross_encoding.py
@@ -19,5 +19,5 @@ def __init__(self, config_path: str, use_sigmoid: bool = True):
  self.use_sigmoid = use_sigmoid
 
  def __call__(self, batch: Batch, config: Optional[EncodingConfig] = None):
- logits = np.squeeze(self.__call__(batch, config))
+ logits = np.squeeze(super().__call__(batch, config))
  return sigmoid(logits) if self.use_sigmoid else logits