DeepSpeed

76ea0534 - Fix missing import in replace_module.py (#2050)

Fix missing import in replace_module.py (#2050) * Fix missing import in replace_module.py * Change import from torch.distributed to deepspeed.comm