chore: simplify dynamo check

2025-05-15 20:28:13 +07:00
parent 64a57ebb62
commit 7fd05c19f7
1 changed files with 12 additions and 13 deletions
--- a/src/axolotl/core/trainer_builder/sft.py
+++ b/src/axolotl/core/trainer_builder/sft.py
@@ -195,8 +195,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
        if self.cfg.greater_is_better:
            training_arguments_kwargs["greater_is_better"] = self.cfg.greater_is_better

-        if self.cfg.torch_compile:
-            if torch._dynamo:  # pylint: disable=protected-access
+        if self.cfg.torch_compile and getattr(torch, "_dynamo", None):
            torch._dynamo.config.suppress_errors = (  # pylint: disable=protected-access
                True
            )