hiyouga 34533b2f35 support vllm
Former-commit-id: d07ad5cc1cdbc13879afd84f653afdfee03a6933
2024-03-07 20:26:31 +08:00
..
2024-03-06 13:14:57 +08:00
2024-03-06 13:14:57 +08:00
2024-03-06 13:14:57 +08:00
2024-03-06 13:14:57 +08:00
2024-03-07 20:26:31 +08:00
2024-03-06 13:14:57 +08:00
2024-03-06 13:14:57 +08:00

Usage:

  • 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)