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

update multigpu/test_qwen2

Browse Source
This commit is contained in:
Sunny Liu
2025-04-21 17:04:17 -04:00
parent 3251b3235f
commit 136407c556
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -28,7 +28,10 @@ class TestMultiGPUQwen2:
cfg = DictDefault( cfg = DictDefault(
{ {
"base_model": base_model, "base_model": base_model,
"load_in_4bit": True, "quantization": {
"backend": "bnb",
"bits": 4,
},
"rl": "dpo", "rl": "dpo",
"chat_template": "chatml", "chat_template": "chatml",
"sequence_len": 2048, "sequence_len": 2048,