From c1bd9c5c0141ec8b981e12232224ee1595cb83d1 Mon Sep 17 00:00:00 2001 From: "D. Ferruzzi" Date: Fri, 1 Nov 2024 17:57:04 -0700 Subject: [PATCH] Bedrock retired the Llama2 model, updated the test to use the closest replacement Llama3 model. (#43600) --- providers/tests/system/amazon/aws/example_bedrock.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/tests/system/amazon/aws/example_bedrock.py b/providers/tests/system/amazon/aws/example_bedrock.py index 3caa9f3a35398..bc28be30174e0 100644 --- a/providers/tests/system/amazon/aws/example_bedrock.py +++ b/providers/tests/system/amazon/aws/example_bedrock.py @@ -63,7 +63,7 @@ SKIP_PROVISION_THROUGHPUT = environ.get("SKIP_RESTRICTED_SYSTEM_TEST_TASKS", default=True) -LLAMA_SHORT_MODEL_ID = "meta.llama2-13b-chat-v1" +LLAMA_SHORT_MODEL_ID = "meta.llama3-8b-instruct-v1:0" TITAN_MODEL_ID = "amazon.titan-text-express-v1:0:8k" TITAN_SHORT_MODEL_ID = TITAN_MODEL_ID.split(":")[0]