We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c90ff8e commit 7481009Copy full SHA for 7481009
1 file changed
sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py
@@ -427,12 +427,12 @@ def _parse_context_length(value) -> int:
427
428
429
def _get_fine_tuning_options_and_model_arn(
430
- model_name: str,
431
- customization_technique: str,
432
- training_type,
433
- sagemaker_session,
434
- sequence_length=None,
435
- hub_name: str = "SageMakerPublicHub"
+ model_name: str,
+ customization_technique: str,
+ training_type,
+ sagemaker_session,
+ sequence_length=None,
+ hub_name: Optional[str] = None
436
) -> tuple:
437
"""Get fine-tuning options and model ARN for given customization technique.
438
0 commit comments