diff --git a/spacy_llm/models/__init__.py b/spacy_llm/models/__init__.py index c1427009..15633845 100644 --- 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", ]