Replace meta-* model with huggingface-* gated model

Author: mollyheamazon

tests/integ/sagemaker/jumpstart/constants.py

@@ -53,6 +53,7 @@ def _to_s3_path(filename: str, s3_prefix: Optional[str]) -> str:
     ("meta-textgeneration-llama-2-7b", "2.*"): ("training-datasets/sec_amazon/"),
     ("meta-textgeneration-llama-2-7b", "3.*"): ("training-datasets/sec_amazon/"),
     ("meta-textgeneration-llama-2-7b", "4.*"): ("training-datasets/sec_amazon/"),
+    ("huggingface-llm-gemma-7b", "*"): ("training-datasets/genuq/small/"),
     ("meta-textgenerationneuron-llama-2-7b", "*"): ("training-datasets/sec_amazon/"),
 }
 

tests/integ/sagemaker/jumpstart/private_hub/estimator/test_jumpstart_private_hub_estimator.py

@@ -38,7 +38,7 @@
 
 TEST_MODEL_IDS = {
     "huggingface-spc-bert-base-cased",
-    "meta-textgeneration-llama-2-7b",
+    "huggingface-llm-gemma-7b",
     "catboost-regression-model",
 }
 
@@ -138,7 +138,7 @@ def test_jumpstart_hub_estimator_with_session(setup, add_model_references):
 
 def test_jumpstart_hub_gated_estimator_with_eula(setup, add_model_references):
 
-    model_id, model_version = "meta-textgeneration-llama-2-7b", "*"
+    model_id, model_version = "huggingface-llm-gemma-7b", "*"
 
     estimator = JumpStartEstimator(
         model_id=model_id,
@@ -172,7 +172,7 @@ def test_jumpstart_hub_gated_estimator_with_eula(setup, add_model_references):
 
 def test_jumpstart_hub_gated_estimator_without_eula(setup, add_model_references):
 
-    model_id, model_version = "meta-textgeneration-llama-2-7b", "*"
+    model_id, model_version = "huggingface-llm-gemma-7b", "*"
 
     estimator = JumpStartEstimator(
         model_id=model_id,

tests/integ/sagemaker/jumpstart/private_hub/model/test_jumpstart_private_hub_model.py

@@ -40,8 +40,8 @@
 TEST_MODEL_IDS = {
     "catboost-classification-model",
     "model-txt2img-stabilityai-stable-diffusion-v2-1-base",
-    "meta-textgeneration-llama-2-7b",
-    "meta-textgeneration-llama-3-2-1b",
+    "huggingface-llm-gemma-7b",
+    "huggingface-llm-gemma-3-1b-instruct",
     "catboost-regression-model",
 }
 
@@ -104,7 +104,7 @@ def test_jumpstart_hub_model_with_default_session(setup, add_model_references):
 
 def test_jumpstart_hub_gated_model(setup, add_model_references):
 
-    model_id = "meta-textgeneration-llama-3-2-1b"
+    model_id = "huggingface-llm-gemma-3-1b-instruct"
 
     model = JumpStartModel(
         model_id=model_id,
@@ -128,7 +128,7 @@ def test_jumpstart_hub_gated_model(setup, add_model_references):
 @pytest.mark.skip(reason="blocking PR checks and release pipeline.")
 def test_jumpstart_gated_model_inference_component_enabled(setup, add_model_references):
 
-    model_id = "meta-textgeneration-llama-3-2-1b"
+    model_id = "huggingface-llm-gemma-3-1b-instruct"
 
     hub_name = os.environ[ENV_VAR_JUMPSTART_SDK_TEST_HUB_NAME]