github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 22:42:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
CZYCW 2023-07-18 10:53:08 +08:00 committed by binmakeswell
parent 77c469e1ba
commit dee1c96344
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
applications/Chat/examples/ray/mmmt_prompt.py
View File

@ -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'])