github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-31 23:35:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Tong Li 2025-03-31 13:14:41 +08:00
parent 27d0108c9f
commit 9d6ede98dc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
applications/ColossalChat/rl_example.py
View File

@ -32,7 +32,7 @@ if __name__ == "__main__":
help="Number of prompts per device. Number of samples = tMbs * num of generation per prompt.",
)
parser.add_argument("-tmbs", "--train-microbatch-size", type=int, default=2, help="Number of samples per device.")
parser.add_argument("-b", "--backend", type=str, default="transformers", choices=["transformers, vllm"])
parser.add_argument("-b", "--backend", type=str, default="transformers", choices=["transformers","vllm"])
parser.add_argument("-a", "--algo", type=str, default="GRPO", choices=["Simple", "GRPO", "EvalGRPO"])
args = parser.parse_args()