From 54673fd6ca39bf86addb20ba7d44a77071b4dc4f Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Thu, 17 Oct 2024 14:12:31 -0400
Subject: [PATCH] also debug if other debug args are set (#1977)

---
 src/axolotl/cli/__init__.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py
index fd5ab3e56..84836bb79 100644
--- 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(