updated to work with newer Vertex AI APIs

PiperOrigin-RevId: 685847928 Change-Id: I7f42f0dfe437931413c823964c8f5145bb0c98ed
google-deepmind · Oct 14, 2024 · c561260 · c561260
1 parent 90619aa
commit c561260
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/concordia/language_model/google_cloud_custom_model.py b/concordia/language_model/google_cloud_custom_model.py
@@ -108,7 +108,7 @@ def sample_text(
 
       try:
         response = (
-            aiplatform.Client()
+            aiplatform.PredictionServiceClient()
             .predict(
                 [{"content": prompt}],
                 parameters={