fix a bug when loading model

hanhainebula · hanhainebula · commit 4bbb986e7ef6 · 2024-07-26T21:56:10.000+08:00
diff --git a/FlagEmbedding/__init__.py b/FlagEmbedding/__init__.py
@@ -1,3 +1,3 @@
-from .flag_models import FlagModel, LLMEmbedder, FlagICLModel
+from .flag_models import FlagModel, LLMEmbedder, FlagICLModel, FlagLLMModel
 from .bge_m3 import BGEM3FlagModel
 from .flag_reranker import FlagReranker, FlagLLMReranker, LayerWiseFlagLLMReranker
