Merge pull request #389 from markstur/crossencoder

CrossEncoderModule with rerank API
caikit · Sep 12, 2024 · 1695c3b · 1695c3b
2 parents 4a027d3 + 8fa67cc
commit 1695c3b
Show file tree

Hide file tree

Showing 3 changed files with 1,225 additions and 0 deletions.
diff --git a/caikit_nlp/modules/text_embedding/__init__.py b/caikit_nlp/modules/text_embedding/__init__.py
@@ -29,4 +29,5 @@
 """
 
 # Local
+from .crossencoder import CrossEncoderModule
 from .embedding import EmbeddingModule