Formatting

kaiko-ai · Feb 19, 2025 · 1dabe91 · 1dabe91
1 parent 3f1e239
commit 1dabe91
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/eva/language/models/__init__.py b/src/eva/language/models/__init__.py
@@ -2,6 +2,6 @@
 
 from eva.language.models import networks, wrappers
 from eva.language.models.networks import TextModule
-from eva.language.models.wrappers import HuggingFaceTextModel
+from eva.language.models.wrappers import HuggingFaceTextModel, LiteLLMTextModel
 
-__all__ = ["networks", "wrappers", "TextModule", "HuggingFaceTextModel"]
+__all__ = ["networks", "wrappers", "TextModule", "HuggingFaceTextModel", "LiteLLMTextModel"]