Skip to content

Commit 4963f3d

Browse files
committed
update tiny artifacts
1 parent f7eb9a9 commit 4963f3d

File tree

6 files changed

+2
-2
lines changed

6 files changed

+2
-2
lines changed

test/sagemaker_tests/huggingface/inference/integration/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424
model_dir = os.path.join(resources_path, "tiny-distilbert-sst-2")
2525
model_dir_sdxl = os.path.join(resources_path, "tiny-sdxl")
26-
model_dir_decoder = os.path.join(resources_path, "tiny-gpt2")
26+
model_dir_decoder = os.path.join(resources_path, "tiny-llama3")
2727
pt_model = "pt_model.tar.gz"
2828
tf_model = "tf_model.tar.gz"
2929
pt_neuron_model = "pt_neuron_model.tar.gz"

test/sagemaker_tests/huggingface/inference/integration/sagemaker/test_neuronx_decoder_hosting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
# instances in the regions corresponding to their availability.
3939
# In future, we would like to configure the logic to run multiple `pytest` commands that can allow
4040
# us to test multiple instances in multiple regions for each image.
41-
@pytest.mark.model("tiny-gpt2")
41+
@pytest.mark.model("tiny-llama3")
4242
@pytest.mark.processor("neuronx")
4343
@pytest.mark.parametrize(
4444
"test_region,test_instance_type",
Binary file not shown.
Binary file not shown.
Binary file not shown.

0 commit comments

Comments
 (0)