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

don't use deepspeed for the fix_untrained_tokens test

Browse Source
This commit is contained in:
Wing Lian
2025-04-06 07:42:11 -04:00
parent 3d0ab75a0c
commit 0cf42ab8a3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/e2e/multigpu/test_llama.py
View File

@@ -894,7 +894,7 @@ class TestMultiGPULlama:
"sample_packing": True, "sample_packing": True,
"bf16": True, "bf16": True,
"save_safetensors": True, "save_safetensors": True,
"deepspeed": str(AXOLOTL_ROOT / "deepspeed_configs/zero1.json"), # "deepspeed": str(AXOLOTL_ROOT / "deepspeed_configs/zero1.json"),
"use_tensorboard": True, "use_tensorboard": True,
} }
) )