Skip to content

Commit e2d9ab3

Browse files
committed
fix test
Former-commit-id: e86f20134b782c8f5c39ead292f8f7582038eb9e
1 parent 3db5cf4 commit e2d9ab3

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

tests/data/processors/test_unsupervised.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,17 +30,18 @@
3030

3131
TRAIN_ARGS = {
3232
"model_name_or_path": TINY_LLAMA,
33-
"stage": "sft",
34-
"do_predict": True,
33+
"stage": "ppo",
34+
"do_train": True,
3535
"finetuning_type": "full",
36-
"eval_dataset": "system_chat",
36+
"reward_model": "",
37+
"reward_model_type": "full",
38+
"dataset": "system_chat",
3739
"dataset_dir": "REMOTE:" + DEMO_DATA,
3840
"template": "llama3",
3941
"cutoff_len": 8192,
4042
"overwrite_cache": True,
4143
"output_dir": "dummy_dir",
4244
"overwrite_output_dir": True,
43-
"predict_with_generate": True,
4445
"fp16": True,
4546
}
4647

0 commit comments

Comments
 (0)