lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix ci

This commit is contained in:
hiyouga 2024-09-05 22:39:47 +08:00
parent 2f939b708f
commit eb5aa9adce
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/e2e/test_train.py
View File

@ -32,7 +32,7 @@ TRAIN_ARGS = {
"dataset_dir": "REMOTE:" + DEMO_DATA, "dataset_dir": "REMOTE:" + DEMO_DATA,
"template": "llama3", "template": "llama3",
"cutoff_len": 1, "cutoff_len": 1,
"overwrite_cache": True, "overwrite_cache": False,
"overwrite_output_dir": True, "overwrite_output_dir": True,
"per_device_train_batch_size": 1, "per_device_train_batch_size": 1,
"max_steps": 1, "max_steps": 1,
@ -55,9 +55,9 @@ OS_NAME = os.environ.get("OS_NAME", "")
[ [
("pt", "c4_demo"), ("pt", "c4_demo"),
("sft", "alpaca_en_demo"), ("sft", "alpaca_en_demo"),
pytest.param("rm", "dpo_en_demo", marks=pytest.mark.xfail(OS_NAME.startswith("windows"), reason="OS error.")),
("dpo", "dpo_en_demo"), ("dpo", "dpo_en_demo"),
("kto", "kto_en_demo"), ("kto", "kto_en_demo"),
pytest.param("rm", "dpo_en_demo", marks=pytest.mark.xfail(OS_NAME.startswith("windows"), reason="OS error.")),
], ],
) )
def test_run_exp(stage: str, dataset: str): def test_run_exp(stage: str, dataset: str):