Add tp-size and pp-size variations to GPT-J model script

Add tp-size and pp-size variations to GPT-J model script #666