Add a specific `device_id` when calling `torch.init_process_group` to avoid NCCL communication errors.#76

gongweibao wants to merge 1 commit intoFoundationVision:mainFoundationVision/Infinity:mainfrom

gongweibao:fixncclinitgongweibao/Infinity:fixncclinitCopy head branch name to clipboard