tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

add pytorch 2.5.0 base images (#1979)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

6
.github/workflows/base.yml vendored
View File

@@ -36,6 +36,12 @@ jobs:
python_version: "3.11" python_version: "3.11"
pytorch: 2.4.1 pytorch: 2.4.1
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX" torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
- cuda: "124"
cuda_version: 12.4.1
cudnn_version: ""
python_version: "3.11"
pytorch: 2.5.0
torch_cuda_arch_list: "7.0 7.5 8.0 8.6 8.7 8.9 9.0+PTX"
steps: steps:
- name: Checkout - name: Checkout
uses: actions/checkout@v3 uses: actions/checkout@v3

2
src/axolotl/cli/__init__.py
View File

@@ -466,7 +466,7 @@ def load_datasets(
cli_args.debug cli_args.debug
or cfg.debug or cfg.debug
or cli_args.debug_text_only 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...") LOG.info("check_dataset_labels...")
check_dataset_labels( check_dataset_labels(

2
src/axolotl/common/cli.py
View File

@@ -23,7 +23,7 @@ class TrainerCliArgs:
debug: bool = field(default=False) debug: bool = field(default=False)
debug_text_only: 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) inference: bool = field(default=False)
merge_lora: bool = field(default=False) merge_lora: bool = field(default=False)
prompter: Optional[str] = field(default=None) prompter: Optional[str] = field(default=None)