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

PiperOrigin-RevId: 620991093
googleapis · Apr 1, 2024 · 1f27c3e · 1f27c3e
1 parent 7f9fab7
commit 1f27c3e
Showing 1 changed file with 1 addition and 0 deletions.
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",
 }