Flax MLM: Allow validation split when loading dataset from local file
# What does this PR do?

In Flax training scripts for MLM, CLM, and T5, this PR enables the option to apply validation-split-percentage when loading datasets from local file. This option already worked when loading standard HF datasets but was missing for local files.


## Who can review?

@patrickvonplaten @patil-suraj 


