in scripts/ft_gemma3n_image_trl.py [0:0]
def prepare_dataset( dataset: DatasetDict, dataset_name: str, dataset_train_split: str