Skip to content

Commit 694b4f2

Browse files
committed
remove references to accept_draft_model_eula
1 parent a7feb54 commit 694b4f2

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

src/sagemaker/jumpstart/model.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -468,8 +468,6 @@ def set_deployment_config(self, config_name: str, instance_type: str) -> None:
468468
instance_type (str):
469469
The instance_type that the model will use after setting
470470
the config.
471-
accept_draft_model_eula (Optional[bool]):
472-
If the config selected comes with a gated additional model data source.
473471
"""
474472
self.__init__(
475473
model_id=self.model_id,

src/sagemaker/serve/builder/jumpstart_builder.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -844,7 +844,6 @@ def _set_additional_model_source(
844844
"""
845845
if speculative_decoding_config:
846846
model_provider = _extract_speculative_draft_model_provider(speculative_decoding_config)
847-
accept_draft_model_eula = speculative_decoding_config.get("AcceptEula", False)
848847

849848
channel_name = _generate_channel_name(self.pysdk_model.additional_model_data_sources)
850849

@@ -874,7 +873,6 @@ def _set_additional_model_source(
874873
self.pysdk_model.set_deployment_config(
875874
config_name=deployment_config.get("DeploymentConfigName"),
876875
instance_type=deployment_config.get("InstanceType"),
877-
accept_draft_model_eula=accept_draft_model_eula,
878876
)
879877
except ValueError as e:
880878
raise ValueError(
@@ -909,7 +907,7 @@ def _set_additional_model_source(
909907
)
910908
else:
911909
self.pysdk_model = _custom_speculative_decoding(
912-
self.pysdk_model, speculative_decoding_config, accept_draft_model_eula
910+
self.pysdk_model, speculative_decoding_config, speculative_decoding_config.get("AcceptEula", False)
913911
)
914912

915913
def _find_compatible_deployment_config(

0 commit comments

Comments
 (0)