if _dataset_path == 'wikitext': _dataset_name = 'wikitext-2-raw-v1' if _dataset_path == 'c4': _dataset_name = 'realnewslike' # Load the dataset print(_dataset_path) #data = load_dataset(data_dir=_dataset_path, name='wikitext-2-raw-v1', split=_split) data = load_datase...