Can `preprocess_data.py` support Huggingface Dataset?

Since there are many datasets in the format of Huggingface datasets, it would be convenient if `preprocess_data.py` can directly preprocess and tokenize from HF datasets.