ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

Published 20 hours ago • verified publisher icon hpcaitech

[BUG]: [chat] Unable to run train_prompts.sh by single card

Open zhang-yi-chi opened this issue 2 years ago • 0 comments

🐛 Describe the bug

Missing definition for prompt_sampler and pretrain_sampler in examples/train_prompts.py when dist.get_world_size() == 1.

Environment

No response

zhang-yi-chi avatar Apr 22 '23 06:04 zhang-yi-chi