[DIPU] Wx/modify maximum schema due to the case in the inference of internlm

[DIPU] Wx/modify maximum schema due to the case in the inference of internlm #4001