diff --git a/fastembed/__init__.py b/fastembed/__init__.py index 906cc1fb..7a2e4134 100644 --- a/fastembed/__init__.py +++ b/fastembed/__init__.py @@ -2,6 +2,7 @@ from fastembed.image import ImageEmbedding from fastembed.late_interaction import LateInteractionTextEmbedding +from fastembed.late_interaction_multimodal import LateInteractionMultimodalEmbedding from fastembed.sparse import SparseEmbedding, SparseTextEmbedding from fastembed.text import TextEmbedding @@ -17,4 +18,5 @@ "SparseEmbedding", "ImageEmbedding", "LateInteractionTextEmbedding", + "LateInteractionMultimodalEmbedding", ] diff --git a/tests/test_late_interaction_multimodal.py b/tests/test_late_interaction_multimodal.py index 0083ee98..b94f0f98 100644 --- a/tests/test_late_interaction_multimodal.py +++ b/tests/test_late_interaction_multimodal.py @@ -3,7 +3,7 @@ from PIL import Image import numpy as np -from fastembed.late_interaction_multimodal import LateInteractionMultimodalEmbedding +from fastembed import LateInteractionMultimodalEmbedding from tests.config import TEST_MISC_DIR