From 7d667f9f8be264bb36e3d720caa0d8210aec5d6b Mon Sep 17 00:00:00 2001 From: Alexey Volkov Date: Wed, 27 Sep 2023 23:34:25 -0700 Subject: [PATCH] feat: LLM - Released `ChatModel` tuning to GA PiperOrigin-RevId: 569078680 --- tests/unit/aiplatform/test_language_models.py | 2 +- vertexai/language_models/_language_models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/aiplatform/test_language_models.py b/tests/unit/aiplatform/test_language_models.py index 325a920d8a..3032c19c10 100644 --- a/tests/unit/aiplatform/test_language_models.py +++ b/tests/unit/aiplatform/test_language_models.py @@ -1557,7 +1557,7 @@ def test_tune_chat_model( _CHAT_BISON_PUBLISHER_MODEL_DICT ), ): - model = preview_language_models.ChatModel.from_pretrained("chat-bison@001") + model = language_models.ChatModel.from_pretrained("chat-bison@001") default_context = "Default context" tuning_job = model.tune_model( diff --git a/vertexai/language_models/_language_models.py b/vertexai/language_models/_language_models.py index c27ab74286..ac68a23a28 100644 --- a/vertexai/language_models/_language_models.py +++ b/vertexai/language_models/_language_models.py @@ -1338,7 +1338,7 @@ def start_chat( ) -class ChatModel(_ChatModelBase): +class ChatModel(_ChatModelBase, _TunableChatModelMixin): """ChatModel represents a language model that is capable of chat. Examples::