patch sentence transformers

michaelfeil · Jun 19, 2023 · 61ebba7 · 61ebba7
1 parent 899a03a
commit 61ebba7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hf_hub_ctranslate2/ct2_sentence_transformers.py b/hf_hub_ctranslate2/ct2_sentence_transformers.py
@@ -62,7 +62,7 @@ def __init__(
         )
 
 
-class CT2Transformer(torch.nn.Module):
+class CT2Transformer(Module):
     """Wrapper around a sentence_transformers.models.Transformer which routes the forward
     call to a CTranslate2 encoder model.