Update src/sagemaker/huggingface/training_compiler/config.py

Lokiiiiii · mchoi8739 · web-flow · commit de3077c59dd4 · 2022-08-22T16:52:35.000-07:00
Co-authored-by: Miyoung &lt;myoung8739@gmail.com&gt;
diff --git a/src/sagemaker/huggingface/training_compiler/config.py b/src/sagemaker/huggingface/training_compiler/config.py
@@ -132,7 +132,7 @@ def validate(
                     if Version(estimator.pytorch_version) in SpecifierSet(">= 1.11"):
                         error_helper_string = (
                             "'pytorchxla' is the only distribution mechanism currently supported "
-                            "for PyTorch >= 1.11 when Training Compiler is enabled. Received "
+                            "for PyTorch >= 1.11 when SageMaker Training Compiler is enabled. Received "
                             "distribution={} which is unsupported."
                         )
                         raise ValueError(error_helper_string.format(estimator.distribution))

Original file line number	Diff line number	Diff line change
`@@ -132,7 +132,7 @@ def validate(`
`132`	`132`	`if Version(estimator.pytorch_version) in SpecifierSet(">= 1.11"):`
`133`	`133`	`error_helper_string = (`
`134`	`134`	`"'pytorchxla' is the only distribution mechanism currently supported "`
`135`		`- "for PyTorch >= 1.11 when Training Compiler is enabled. Received "`
	`135`	`+ "for PyTorch >= 1.11 when SageMaker Training Compiler is enabled. Received "`
`136`	`136`	`"distribution={} which is unsupported."`
`137`	`137`	`)`
`138`	`138`	`raise ValueError(error_helper_string.format(estimator.distribution))`