Skip to content

fix(serve): remove IC data-source collapse hack from recommendation deploy - #6101

Open
ZealSV wants to merge 2 commits into
aws:masterfrom
ZealSV:remove-ic-sdkt-hack
Open

fix(serve): remove IC data-source collapse hack from recommendation deploy#6101
ZealSV wants to merge 2 commits into
aws:masterfrom
ZealSV:remove-ic-sdkt-hack

Conversation

@ZealSV

@ZealSV ZealSV commented Jul 24, 2026

Copy link
Copy Markdown
Contributor

Inference Components now support AdditionalModelDataSources natively (kernel tuning / speculative decoding channels), so _deploy_recommendation no longer needs to collapse an optimized recommendation's base_model + draft channels into a single primary ModelDataSource. Deploy the recommendation's ModelPackage directly and let the hosting stack resolve the channels.

Removes the base_model-promotion / OPTION_SPECULATIVE_DRAFT_MODEL rewiring block and its now-unused shape imports (AdditionalModelDataSource, ModelDataSource, S3ModelDataSource, SPECULATIVE_DRAFT_MODEL). Updates the speculative-decoding unit tests to assert the ModelPackage pass-through.

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Comment thread sagemaker-serve/src/sagemaker/serve/model_builder.py
…eploy

Inference Components now support AdditionalModelDataSources natively (kernel
tuning / speculative decoding channels), so _deploy_recommendation no longer
needs to collapse an optimized recommendation's base_model + draft channels
into a single primary ModelDataSource. Deploy the recommendation's
ModelPackage directly and let the hosting stack resolve the channels.

Removes the base_model-promotion / OPTION_SPECULATIVE_DRAFT_MODEL rewiring
block and its now-unused shape imports (AdditionalModelDataSource,
ModelDataSource, S3ModelDataSource, SPECULATIVE_DRAFT_MODEL). Updates the
speculative-decoding unit tests to assert the ModelPackage pass-through.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants