Fix PaLM import in __init__.

explosion · Oct 5, 2023 · 1ad50a1 · 1ad50a1
1 parent d891513
commit 1ad50a1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spacy_llm/models/__init__.py b/spacy_llm/models/__init__.py
@@ -1,6 +1,6 @@
 from .hf import dolly_hf, openllama_hf, stablelm_hf
 from .langchain import query_langchain
-from .rest import anthropic, cohere, noop, openai
+from .rest import anthropic, cohere, noop, openai, palm
 
 __all__ = [
     "anthropic",
@@ -10,5 +10,6 @@
     "noop",
     "stablelm_hf",
     "openllama_hf",
+    "palm",
     "query_langchain",
 ]