Skip to content

Commit f721184

Browse files
author
Jonathan Makunga
committed
Refactoring
1 parent 1d1233d commit f721184

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/serve/builder/jumpstart_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -668,9 +668,9 @@ def _build_for_jumpstart(self):
668668
def _optimize_for_jumpstart(
669669
self,
670670
output_path: Optional[str] = None,
671+
instance_type: Optional[str] = None,
671672
tags: Optional[Tags] = None,
672673
job_name: Optional[str] = None,
673-
instance_type: Optional[str] = None,
674674
accept_eula: Optional[bool] = None,
675675
quantization_config: Optional[Dict] = None,
676676
compilation_config: Optional[Dict] = None,
@@ -684,9 +684,9 @@ def _optimize_for_jumpstart(
684684
685685
Args:
686686
output_path (Optional[str]): Specifies where to store the compiled/quantized model.
687+
instance_type (str): Target deployment instance type that the model is optimized for.
687688
tags (Optional[Tags]): Tags for labeling a model optimization job. Defaults to ``None``.
688689
job_name (Optional[str]): The name of the model optimization job. Defaults to ``None``.
689-
instance_type (str): Target deployment instance type that the model is optimized for.
690690
accept_eula (bool): For models that require a Model Access Config, specify True or
691691
False to indicate whether model terms of use have been accepted.
692692
The `accept_eula` value must be explicitly defined as `True` in order to

0 commit comments

Comments
 (0)