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

Lokiiiiii · mchoi8739 · web-flow · commit 1db9ed088471 · 2022-08-22T16:52:09.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
@@ -123,7 +123,7 @@ def validate(
                     if Version(estimator.pytorch_version) in SpecifierSet("< 1.11"):
                         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 "
                             "pytorch_version={} which is unsupported."
                         )
                         raise ValueError(error_helper_string.format(estimator.pytorch_version))

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ def validate(`
`123`	`123`	`if Version(estimator.pytorch_version) in SpecifierSet("< 1.11"):`
`124`	`124`	`error_helper_string = (`
`125`	`125`	`"Distribution mechanism 'pytorchxla' is currently only supported for "`
`126`		`- "PyTorch >= 1.11 when Training Compiler is enabled. Received "`
	`126`	`+ "PyTorch >= 1.11 when SageMaker Training Compiler is enabled. Received "`
`127`	`127`	`"pytorch_version={} which is unsupported."`
`128`	`128`	`)`
`129`	`129`	`raise ValueError(error_helper_string.format(estimator.pytorch_version))`