* [testing] add beit model * [beit] fix bugs * [beit] fix bugs * [testing] fix bugs
[colotensor] add megatron initialization for gpt2