Skip to content
Snippets Groups Projects
Unverified Commit 455f529a authored by akumar's avatar akumar Committed by GitHub
Browse files

Adding Support for MedLM Models (#11911)


* Adding Support for MedLM Models

* cr

---------

Co-authored-by: default avatarHaotian Zhang <socool.king@gmail.com>
parent 5e238d5b
No related branches found
No related tags found
No related merge requests found
......@@ -19,7 +19,13 @@ from vertexai.language_models import InputOutputTextPair
from llama_index.core.base.llms.types import ChatMessage, MessageRole
CHAT_MODELS = ["chat-bison", "chat-bison-32k", "chat-bison@001"]
TEXT_MODELS = ["text-bison", "text-bison-32k", "text-bison@001"]
TEXT_MODELS = [
"text-bison",
"text-bison-32k",
"text-bison@001",
"medlm-medium",
"medlm-large",
]
CODE_MODELS = ["code-bison", "code-bison-32k", "code-bison@001"]
CODE_CHAT_MODELS = ["codechat-bison", "codechat-bison-32k", "codechat-bison@001"]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment