From de4ecb48c78abcf05ff5c79c4469f52343931259 Mon Sep 17 00:00:00 2001 From: George Date: Wed, 12 Feb 2025 18:10:07 +0100 Subject: [PATCH] new: add late interaction multimodal to package init (#471) --- fastembed/__init__.py | 2 ++ tests/test_late_interaction_multimodal.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/fastembed/__init__.py b/fastembed/__init__.py index 906cc1fb7..7a2e41340 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 0083ee985..b94f0f981 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