tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: remove deprecated clause

Browse Source
This commit is contained in:
NanoCode012
2025-05-14 16:55:34 +07:00
parent 316b450a87
commit 8a336a2c33
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/core/trainers/trainer_builder/sft.py
View File

@@ -196,9 +196,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
training_arguments_kwargs["greater_is_better"] = self.cfg.greater_is_better training_arguments_kwargs["greater_is_better"] = self.cfg.greater_is_better
if self.cfg.torch_compile: if self.cfg.torch_compile:
if torch.__version__ < "2.1.0": # pylint: disable=protected-access if torch._dynamo: # pylint: disable=protected-access
LOG.warning("torch>=2.1.0 required for torch_compile to work properly")
elif torch._dynamo: # pylint: disable=protected-access
torch._dynamo.config.suppress_errors = ( # pylint: disable=protected-access torch._dynamo.config.suppress_errors = ( # pylint: disable=protected-access
True True
) )