We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3d36503 commit c5747adCopy full SHA for c5747ad
1 file changed
sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py
@@ -338,12 +338,12 @@ def _parse_context_length(value) -> int:
338
339
340
def _get_fine_tuning_options_and_model_arn(
341
- model_name: str,
342
- customization_technique: str,
343
- training_type,
344
- sagemaker_session,
345
- sequence_length=None,
346
- hub_name: str = "SageMakerPublicHub"
+ model_name: str,
+ customization_technique: str,
+ training_type,
+ sagemaker_session,
+ sequence_length=None,
+ hub_name: Optional[str] = None
347
) -> tuple:
348
"""Get fine-tuning options and model ARN for given customization technique.
349
0 commit comments