Update src/sagemaker/huggingface/estimator.py

Lokiiiiii · mchoi8739 · web-flow · commit 33b317417850 · 2022-08-22T16:52:50.000-07:00
Co-authored-by: Miyoung &lt;myoung8739@gmail.com&gt;
diff --git a/src/sagemaker/huggingface/estimator.py b/src/sagemaker/huggingface/estimator.py
@@ -223,7 +223,7 @@ def __init__(
         elif distribution is not None and "pytorchxla" in distribution:
             raise ValueError(
                 "Distributed training through PyTorch XLA is currently only supported "
-                "when Training Compiler is enabled."
+                "when SageMaker Training Compiler is enabled. To learn more, see Enable SageMaker Training Compiler at https://docs.aws.amazon.com/sagemaker/latest/dg/training-compiler-enable.html."
             )
         self.compiler_config = compiler_config
 

Original file line number	Diff line number	Diff line change
`@@ -223,7 +223,7 @@ def __init__(`
`223`	`223`	`elif distribution is not None and "pytorchxla" in distribution:`
`224`	`224`	`raise ValueError(`
`225`	`225`	`"Distributed training through PyTorch XLA is currently only supported "`
`226`		`- "when Training Compiler is enabled."`
	`226`	`+ "when SageMaker Training Compiler is enabled. To learn more, see Enable SageMaker Training Compiler at https://docs.aws.amazon.com/sagemaker/latest/dg/training-compiler-enable.html."`
`227`	`227`	`)`
`228`	`228`	`self.compiler_config = compiler_config`
`229`	`229`