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

make sure to download fixtures for kd test (#2541)

Browse Source 
* make sure to download fixtures for kd test

* use same alpaca dataset
This commit is contained in:
Wing Lian
2025-04-21 10:31:50 -04:00
committed by GitHub
parent 341e95aac9
commit 7651550850
6 changed files with 36 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/e2e/multigpu/solo/__init__.py
View File

@@ -0,0 +1,2 @@
# Tests under this directory should get run "solo" on their own as they
# seem to cause issues when run in the same batch as other tests.

3
tests/e2e/multigpu/solo/test_flex.py
View File

@@ -49,8 +49,9 @@ class TestPackedFlex:
},
"datasets": [
{
"path": "vicgalle/alpaca-gpt4",
"path": "tatsu-lab/alpaca",
"type": "alpaca",
"split": "train[:10%]",
},
],
"num_epochs": 1,

3
tests/e2e/patched/test_resume.py
View File

@@ -46,8 +46,9 @@ class TestResumeLlama:
},
"datasets": [
{
"path": "vicgalle/alpaca-gpt4",
"path": "tatsu-lab/alpaca",
"type": "alpaca",
"split": "train[:10%]",
},
],
"num_epochs": 2,

3
tests/e2e/solo/test_flex.py
View File

@@ -41,8 +41,9 @@ class TestPackedFlex(unittest.TestCase):
},
"datasets": [
{
"path": "vicgalle/alpaca-gpt4",
"path": "tatsu-lab/alpaca",
"type": "alpaca",
"split": "train[:10%]",
},
],
"num_epochs": 1,

3
tests/e2e/test_packing_loss.py
View File

@@ -40,8 +40,9 @@ class TestPackedLlama(unittest.TestCase):
},
"datasets": [
{
"path": "vicgalle/alpaca-gpt4",
"path": "tatsu-lab/alpaca",
"type": "alpaca",
"split": "train[:10%]",
},
],
"num_epochs": 1,