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 668064b commit 2083bacCopy full SHA for 2083bac
src/sagemaker/huggingface/training_compiler/config.py
@@ -103,7 +103,9 @@ def validate(cls, estimator):
103
super(TrainingCompilerConfig, cls).validate(estimator)
104
105
if estimator.pytorch_version:
106
- if (Version(estimator.pytorch_version) in SpecifierSet("< 1.9")) or (Version(estimator.pytorch_version) in SpecifierSet("> 1.11")):
+ if (Version(estimator.pytorch_version) in SpecifierSet("< 1.9")) or (
107
+ Version(estimator.pytorch_version) in SpecifierSet("> 1.11")
108
+ ):
109
error_helper_string = (
110
"Training Compiler is only valid between HuggingFace PyTorch 1.9-1.11 "
111
"for SageMaker Training Compiler."
0 commit comments