Fix PT TF ViTMAE
# What does this PR do?

Fix PT TF ViTMAE: just use some settings both in PT/TF (instead of in only one model). Otherwise, the PT/TF equivalence tests for them won't use something like `std = 0.02` , and gets larger (init) weights --> larger diff in outputs. 

Also, **the `eps` for `layer norm` layers should be the same in PT/TF**.
(not a real big deal in practice, since here is `1e-5` v.s. `1e-12` -> but it also affects the tests)
Let's merge since CI fails quite a lot on this one<|||||>> Let's merge since CI fails quite a lot on this one

Ok! 