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

support disable shuffling

Browse Source 
Former-commit-id: 9d8c35fd6b838ede0bd6827c6c6121f2cba2b11b
This commit is contained in:
hiyouga
2024-12-19 08:53:21 +00:00
parent eca06531c3
commit 01eeae50b5
9 changed files with 139 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/e2e/test_train.py
View File

@@ -60,12 +60,12 @@ OS_NAME = os.getenv("OS_NAME", "")
],
)
def test_run_exp(stage: str, dataset: str):
output_dir = os.path.join("output", f"train_{stage}")
output_dir = os.path.join("output", f"dummy_dir/train_{stage}")
run_exp({"stage": stage, "dataset": dataset, "output_dir": output_dir, **TRAIN_ARGS})
assert os.path.exists(output_dir)
def test_export():
export_dir = os.path.join("output", "llama3_export")
export_dir = os.path.join("output", "dummy_dir/llama3_export")
export_model({"export_dir": export_dir, **INFER_ARGS})
assert os.path.exists(export_dir)