From 8a336a2c337203c1f04bcaaf88bd4546e660d0e1 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Wed, 14 May 2025 16:55:34 +0700 Subject: [PATCH] fix: remove deprecated clause --- src/axolotl/core/trainers/trainer_builder/sft.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/axolotl/core/trainers/trainer_builder/sft.py b/src/axolotl/core/trainers/trainer_builder/sft.py index 8ba1f5f20..5caf075a3 100644 --- a/src/axolotl/core/trainers/trainer_builder/sft.py +++ b/src/axolotl/core/trainers/trainer_builder/sft.py @@ -196,9 +196,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): training_arguments_kwargs["greater_is_better"] = self.cfg.greater_is_better if self.cfg.torch_compile: - if torch.__version__ < "2.1.0": # pylint: disable=protected-access - LOG.warning("torch>=2.1.0 required for torch_compile to work properly") - elif torch._dynamo: # pylint: disable=protected-access + if torch._dynamo: # pylint: disable=protected-access torch._dynamo.config.suppress_errors = ( # pylint: disable=protected-access True )