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

Update dependencies and show slow tests in CI (#2492)

Browse Source 
* use latest torchao, gradio, schedule-free

* get info on slow tests

* speed up tests by avoiding gradient checkpointing and reducing eval size
This commit is contained in:
Wing Lian
2025-04-05 17:41:31 -04:00
committed by GitHub
parent 949471039f
commit e7e0cd97ce
4 changed files with 24 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/e2e/multigpu/test_qwen2.py
View File

@@ -37,7 +37,7 @@ class TestMultiGPUQwen2:
"lora_alpha": 16,
"lora_dropout": 0.05,
"lora_target_linear": True,
"val_set_size": 0.05,
"val_set_size": 0.01,
"datasets": [
{
"path": "Intel/orca_dpo_pairs",
@@ -57,7 +57,7 @@ class TestMultiGPUQwen2:
"flash_attention": True,
"bf16": "auto",
"tf32": True,
"gradient_checkpointing": True,
# "gradient_checkpointing": True,
"gradient_checkpointing_kwargs": {
"use_reentrant": False,
},