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
This commit is contained in:
Wing Lian
2024-10-18 03:36:51 -04:00
committed by GitHub
parent f62e23737b
commit 67f744dc8c
3 changed files with 8 additions and 2 deletions

View File

@@ -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(

View File

@@ -23,7 +23,7 @@ class TrainerCliArgs:
debug: bool = field(default=False)
debug_text_only: bool = field(default=False)
debug_num_examples: int = field(default=5)
debug_num_examples: int = field(default=0)
inference: bool = field(default=False)
merge_lora: bool = field(default=False)
prompter: Optional[str] = field(default=None)