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]