support vllm
This commit is contained in:
1 parent
f74f804a71
commit
d07ad5cc1c
32 files changed
+752
-316
No files matched your search
@@ -1,5 +1,8 @@
|
||||
Usage:
|
||||
|
||||
- `pretrain.sh`
|
||||
- `sft.sh` -> `reward.sh` -> `ppo.sh`
|
||||
- `sft.sh` -> `dpo.sh` -> `predict.sh`
|
||||
- `pretrain.sh`: do pre-train (optional)
|
||||
- `sft.sh`: do supervised fine-tune
|
||||
- `reward.sh`: do reward modeling (must after sft.sh)
|
||||
- `ppo.sh`: do PPO training (must after sft.sh and reward.sh)
|
||||
- `dpo.sh`: do DPO training (must after sft.sh)
|
||||
- `predict.sh`: do predict (must after sft.sh and dpo.sh)
|
||||
Reference in new issue
Block a user