add pytorch 2.5.0 base images (#1979)

* add pytorch 2.5.0 base images * make sure num examples for debug is zero and fix comparison
2024-10-18 03:36:51 -04:00
parent f62e23737b
commit 67f744dc8c
3 changed files with 8 additions and 2 deletions
--- a/src/axolotl/cli/init.py
+++ b/src/axolotl/cli/init.py
@@ -466,7 +466,7 @@ def load_datasets(
        cli_args.debug
        or cfg.debug
        or cli_args.debug_text_only
-        or cli_args.debug_num_examples
+        or int(cli_args.debug_num_examples) > 0
    ):
        LOG.info("check_dataset_labels...")
        check_dataset_labels(