Skip to content

Commit

Permalink
removed unneccesary cloud client setup
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 687472624
Change-Id: Id8d2cfe45d5797ab2955854143e2f71645d4a0e6
  • Loading branch information
minsukchang authored and copybara-github committed Oct 19, 2024
1 parent 838696e commit c7e8777
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions concordia/language_model/google_cloud_custom_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ def __init__(
"""Initializes the instance.
Args:
model_name: The language model to use. E.g., "gemma2-9b-it"
model_name: The endpoint ID of the language model to use.
project: Your Google Cloud project ID.
location: The region where the model is deployed.
measurements: The measurements object to log usage statistics to.
Expand All @@ -80,7 +80,6 @@ def __init__(
self._location = location
self._measurements = measurements
self._channel = channel
aiplatform.init(model_name=model_name, project=project, location=location)

@override
# sample_text:
Expand Down

0 comments on commit c7e8777

Please sign in to comment.