[Flax tests/FlaxBert] make from_pretrained test faster (#15561)
[Flax tests/FlaxBert] make from_pretrained test faster (#15561)


huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers