Update src/axolotl/common/datasets.py

Co-authored-by: Dan Saunders <danjsaund@gmail.com>
2025-04-09 09:36:05 -04:00
parent c12906134d
commit f8e92407ff
1 changed files with 1 additions and 3 deletions
--- a/src/axolotl/common/datasets.py
+++ b/src/axolotl/common/datasets.py
@@ -129,9 +129,7 @@ def load_preference_datasets(
        total_num_steps = None

    if cli_args.debug or cfg.debug:
-        if cfg.rl == "grpo":
-            pass
-        else:
+        if not cfg.rl == "grpo":
            LOG.info("check_dataset_labels...")

            tokenizer = load_tokenizer(cfg)