DeepSpeed

a7ee688a - Update replace_module.py, test-gptj.py related fix (#2269)

Update replace_module.py, test-gptj.py related fix (#2269) Fix RuntimeError: Boolean value of Tensor with more than one value is ambiguous when running test-gptj.py