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

Lokiiiiii · mchoi8739 · web-flow · commit 192437c96c95 · 2022-08-22T16:52:40.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
@@ -115,7 +115,7 @@ def validate(
                 if estimator.tensorflow_version:
                     error_helper_string = (
                         "Distribution mechanism 'pytorchxla' is currently only supported for "
-                        "PyTorch >= 1.11 when Training Compiler is enabled. Received "
+                        "PyTorch >= 1.11 when SageMaker Training Compiler is enabled. Received "
                         "tensorflow_version={} which is unsupported."
                     )
                     raise ValueError(error_helper_string.format(estimator.tensorflow_version))

Original file line number	Diff line number	Diff line change
`@@ -115,7 +115,7 @@ def validate(`
`115`	`115`	`if estimator.tensorflow_version:`
`116`	`116`	`error_helper_string = (`
`117`	`117`	`"Distribution mechanism 'pytorchxla' is currently only supported for "`
`118`		`- "PyTorch >= 1.11 when Training Compiler is enabled. Received "`
	`118`	`+ "PyTorch >= 1.11 when SageMaker Training Compiler is enabled. Received "`
`119`	`119`	`"tensorflow_version={} which is unsupported."`
`120`	`120`	`)`
`121`	`121`	`raise ValueError(error_helper_string.format(estimator.tensorflow_version))`