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

RezaYazdaniAminabadi merged 1 commit intodeepspeedai:masterdeepspeedai/DeepSpeed:masterfrom

molly-smith:gptj-fixmolly-smith/DeepSpeed:gptj-fixCopy head branch name to clipboard