super

huggingface · Mar 27, 2024 · 0ff7607 · 0ff7607
1 parent 10a6fce
commit 0ff7607
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/setfit/modeling.py b/src/setfit/modeling.py
@@ -218,6 +218,7 @@ def __init__(
         sentence_transformers_kwargs: Optional[Dict] = None,
         **kwargs,
     ) -> None:
+        super(SetFitModel, self).__init__()
         self.model_body = model_body
         self.model_head = model_head
         self.multi_target_strategy = multi_target_strategy