Skip to content

Commit 579e359

Browse files
author
Joseph Zhang
committed
Restore default instance type. Tweak integ test.
1 parent ca9e014 commit 579e359

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,7 @@ class ModelBuilder(Triton, DJL, JumpStart, TGI, Transformers, TensorflowServing,
274274
metadata={"help": "Define the s3 location where you want to upload the model package"},
275275
)
276276
instance_type: Optional[str] = field(
277-
default=None,
277+
default="ml.c5.xlarge",
278278
metadata={"help": "Define the instance_type of the endpoint"},
279279
)
280280
schema_builder: Optional[SchemaBuilder] = field(

tests/integ/sagemaker/serve/test_schema_builder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,9 @@
3434

3535
def test_model_builder_happy_path_with_only_model_id_text_generation(sagemaker_session):
3636
model_builder = ModelBuilder(
37-
model="HuggingFaceH4/zephyr-7b-beta", sagemaker_session=sagemaker_session
37+
model="HuggingFaceH4/zephyr-7b-beta",
38+
sagemaker_session=sagemaker_session,
39+
instance_type=None,
3840
)
3941

4042
model = model_builder.build(sagemaker_session=sagemaker_session)

0 commit comments

Comments
 (0)