feat: LLM - Add RLHF-tuning support for text-bison@002

vertex-sdk-bot · copybara-github · commit 1f27c3eb5bb9 · 2024-04-01T16:53:04.000-07:00
PiperOrigin-RevId: 620991093
diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py
@@ -3448,6 +3448,7 @@ class _PreviewCodeGenerationModel(CodeGenerationModel, _CountTokensCodeGeneratio
 _SUPPORTED_RLHF_MODELS = {
     "text-bison@001",
     "chat-bison@001",
+    "text-bison@002",
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -3448,6 +3448,7 @@ class _PreviewCodeGenerationModel(CodeGenerationModel, _CountTokensCodeGeneratio`
`3448`	`3448`	`_SUPPORTED_RLHF_MODELS = {`
`3449`	`3449`	`"text-bison@001",`
`3450`	`3450`	`"chat-bison@001",`
	`3451`	`+ "text-bison@002",`
`3451`	`3452`	`}`
`3452`	`3453`
`3453`	`3454`