accelerate

c4460e33 - fix: specify device_ids in torch.distributed.barrier for PartialState (#3744)