also debug if other debug args are set (#1977)

2024-10-17 14:12:31 -04:00
parent 6d9a3c4d81
commit 54673fd6ca
1 changed files with 6 additions and 1 deletions
--- a/src/axolotl/cli/init.py
+++ b/src/axolotl/cli/init.py
@@ -462,7 +462,12 @@ def load_datasets(
        processor=processor,
    )

-    if cli_args.debug or cfg.debug:
+    if (
+        cli_args.debug
+        or cfg.debug
+        or cli_args.debug_text_only
+        or cli_args.debug_num_examples
+    ):
        LOG.info("check_dataset_labels...")
        check_dataset_labels(
            train_dataset.select(