Set proper arguments when constructing models in unit tests

[xinyu-intel]

when contructing models in unit test, arguments need to be set. Take ParallelMLP() as an example, get_args() is used to get some args setted by user in the constructor func:

Megatron-DeepSpeed/megatron/model/transformer.py

Lines 99 to 101 in ea4b67a

	def __init__(self, config, moe=False, enable_expert_tensor_parallelism=False):
	super(ParallelMLP, self).__init__()
	args = get_args()

So, need to set_args() accordingly.

[xinyu-intel]

cc @polisettyvarma

[costin-eseanu]

Everything looks good, approved.

[xinyu-intel]

@tjruwase hi, can you please help on review?

[tjruwase]

@tjruwase hi, can you please help on review?

@xinyu-intel, @costin-eseanu has already approved and will merge. Thanks!