diff --git a/tests/local_testing/test_streaming.py b/tests/local_testing/test_streaming.py index 0d9f84a301..f63ec3859f 100644 --- a/tests/local_testing/test_streaming.py +++ b/tests/local_testing/test_streaming.py @@ -1318,7 +1318,7 @@ async def test_completion_replicate_llama3_streaming(sync_mode): # ["bedrock/cohere.command-r-plus-v1:0", None], ["anthropic.claude-3-sonnet-20240229-v1:0", None], # ["mistral.mistral-7b-instruct-v0:2", None], - ["bedrock/amazon.titan-tg1-large", None], + ["bedrock/amazon.titan-text-express-v1", None], # ["meta.llama3-8b-instruct-v1:0", None], ], )