We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent df711e3 commit 8f886feCopy full SHA for 8f886fe
src/sagemaker/jumpstart/factory/estimator.py
@@ -917,7 +917,7 @@ def _add_config_name_to_kwargs(
917
model_version=kwargs.model_version,
918
sagemaker_session=kwargs.sagemaker_session,
919
scope=JumpStartScriptScope.TRAINING,
920
- model_type=kwargs.model_type,
+ model_type=kwargs.model_type[0],
921
tolerate_deprecated_model=kwargs.tolerate_deprecated_model,
922
tolerate_vulnerable_model=kwargs.tolerate_vulnerable_model,
923
hub_arn=kwargs.hub_arn,
0 commit comments