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

chore: simplify dynamo check

Browse Source
This commit is contained in:
NanoCode012
2025-05-15 20:28:13 +07:00
parent 64a57ebb62
commit 7fd05c19f7
1 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/core/trainer_builder/sft.py
View File

@@ -195,8 +195,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
if self.cfg.greater_is_better: if self.cfg.greater_is_better:
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 and getattr(torch, "_dynamo", None):
if 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
) )