Fix validation for early stopping (#1358)

2024-03-04 08:45:18 +05:30
parent 170d4d7092
commit b5b44925ec
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/utils/config/models/input/v0_4_1/init.py
+++ b/src/axolotl/utils/config/models/input/v0_4_1/init.py
@@ -822,7 +822,7 @@ class AxolotlInputConfig(
    @model_validator(mode="after")
    def check_early_stopping(self):
        if self.early_stopping_patience:
-            if not self.save_steps or self.eval_steps:
+            if not self.save_steps or not self.eval_steps:
                raise ValueError(
                    "`early_stopping_patience` requires save_steps and eval_steps to be set. eval_steps should evenly divide save_steps."
                )