Update __init__.py

explosion · Oct 24, 2023 · 548ce74 · 548ce74
1 parent caecd85
commit 548ce74
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spacy_llm/models/__init__.py b/spacy_llm/models/__init__.py
@@ -1,7 +1,7 @@
 from .hf import dolly_hf, openllama_hf, stablelm_hf
 from .langchain import query_langchain
 from .rest import anthropic, cohere, noop, openai, palm
-
+from .bedrock import titan_lite, titan_express
 __all__ = [
     "anthropic",
     "cohere",
@@ -12,4 +12,6 @@
     "openllama_hf",
     "palm",
     "query_langchain",
+    "titan_lite", 
+    "titan_express",
 ]