From 61ebba75f4fa26dae748915395ebad309fbb7db5 Mon Sep 17 00:00:00 2001
From: Michael Feil <michael.feil@michaelfeil.eu>
Date: Mon, 19 Jun 2023 09:04:46 +0200
Subject: [PATCH] patch sentence transformers

---
 hf_hub_ctranslate2/ct2_sentence_transformers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hf_hub_ctranslate2/ct2_sentence_transformers.py b/hf_hub_ctranslate2/ct2_sentence_transformers.py
index 377d034..ce24268 100644
--- 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.