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

### 🐛 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_