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

gongweibao wants to merge 1 commit into

FoundationVision:mainFoundationVision/Infinity:mainfrom

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