Skip to content

Commit dbc5b7b

Browse files
Update bedrock.py (#665)
1 parent 71418f9 commit dbc5b7b

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

src/backend/model_deployments/bedrock.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,6 @@ class BedrockDeployment(BaseDeployment):
3333

3434
def __init__(self, **kwargs: Any):
3535
self.client = cohere.BedrockClient(
36-
# TODO: remove hardcoded models once the SDK is updated
37-
chat_model="cohere.command-r-plus-v1:0",
38-
embed_model="cohere.embed-multilingual-v3",
39-
generate_model="cohere.command-text-v14",
4036
aws_access_key=get_model_config_var(
4137
BEDROCK_ACCESS_KEY_ENV_VAR, BedrockDeployment.access_key, **kwargs
4238
),

0 commit comments

Comments
 (0)