Merge pull request #1068 from ycjcl868/patch-2

staoxiao · web-flow · commit 06a2113c5351 · 2024-08-21T00:33:27.000+08:00
fix: __init__.py not export LightWeightFlagLLMReranker
diff --git a/FlagEmbedding/__init__.py b/FlagEmbedding/__init__.py
@@ -1,3 +1,3 @@
 from .flag_models import FlagModel, LLMEmbedder, FlagICLModel, FlagLLMModel
 from .bge_m3 import BGEM3FlagModel
-from .flag_reranker import FlagReranker, FlagLLMReranker, LayerWiseFlagLLMReranker
+from .flag_reranker import FlagReranker, FlagLLMReranker, LayerWiseFlagLLMReranker, LightWeightFlagLLMReranker
