Skip to content

GitLab

Commit dee1c963 authored by CZYCW's avatar CZYCW Committed by binmakeswell
Browse files

[NFC] policy applications/Chat/examples/ray/mmmt_prompt.py code style (#4250)

parent 77c469e1
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 8 deletions
applications/Chat/examples/ray/mmmt_prompt.py
View file @ dee1c963
...@@ -161,11 +161,9 @@ if __name__ == '__main__': ...@@ -161,11 +161,9 @@ if __name__ == '__main__':
parser.add_argument('--prompt_path', type=str, default=None) parser.add_argument('--prompt_path', type=str, default=None)
parser.add_argument('--num_makers', type=int, default=1) parser.add_argument('--num_makers', type=int, default=1)
parser.add_argument('--num_trainers', type=int, default=1) parser.add_argument('--num_trainers', type=int, default=1)
parser.add_argument('--trainer_strategy', parser.add_argument(
choices=[ '--trainer_strategy',
'ddp', 'colossalai_gemini', 'colossalai_zero2', 'colossalai_gemini_cpu', choices=['ddp', 'colossalai_gemini', 'colossalai_zero2', 'colossalai_gemini_cpu', 'colossalai_zero2_cpu'],
'colossalai_zero2_cpu'
],
default='ddp') default='ddp')
parser.add_argument('--maker_strategy', choices=['naive'], default='naive') parser.add_argument('--maker_strategy', choices=['naive'], default='naive')
parser.add_argument('--model', default='gpt2', choices=['gpt2', 'bloom', 'opt', 'llama']) parser.add_argument('--model', default='gpt2', choices=['gpt2', 'bloom', 'opt', 'llama'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment