From a5b2d62dd1401490550f3e7e8df01cfec2a5e99f Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Thu, 23 Jan 2025 13:26:59 -0800 Subject: [PATCH] fix: jumpstart model hub test --- .../private_hub/model/test_jumpstart_private_hub_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py b/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py index 751162d2e6..d9eb39db95 100644 --- a/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py +++ b/tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py @@ -107,7 +107,7 @@ def test_jumpstart_hub_gated_model(setup, add_model_references): def test_jumpstart_gated_model_inference_component_enabled(setup, add_model_references): - model_id = "meta-textgeneration-llama-2-7b" + model_id = "meta-textgeneration-llama-3-2-1b" hub_name = os.environ[ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME]