Update registry handle.

explosion · Sep 26, 2023 · 11e8a3a · 11e8a3a
1 parent 44c191e
commit 11e8a3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spacy_llm/models/rest/palm/registry.py b/spacy_llm/models/rest/palm/registry.py
@@ -7,7 +7,7 @@
 from .model import Endpoints, PaLM
 
 
-@registry.llm_models("spacy.ChatBison.v1")
+@registry.llm_models("spacy.Bison.v1")
 def palm_bison(
     config: Dict[Any, Any] = SimpleFrozenDict(temperature=0),
     name: Literal["chat-bison-001", "text-bison-001"] = "text-bison-001",  # noqa: F821