Get error when want to use Colossal-LLaMA-2

[zhpacer]

CUDA: Build cuda_11.8.r11.8/compiler.31833905_0
following: https://github.com/hpcaitech/ColossalAI/tree/main/applications/Colossal-LLaMA-2
Error:
utils/flash_attention_patch.py", line 22, in
from colossalai.accelerator import get_accelerator
ModuleNotFoundError: No module named 'colossalai.accelerator'

[zhpacer]

after copy the accelerator dir to the installed colossalai dir, can continue, but then get another error:
File "/opt/conda/lib/python3.8/site-packages/colossalai/booster/plugin/dp_plugin_base.py", line 63, in prepare_dataloader
return DataLoader(dataset,
TypeError: init() got an unexpected keyword argument 'distributed_sampler_cls'

seems, the version of colossalai is not right?