model_name_or_path: the name of the model on the Hub or a local path to the pretrained model --dataset_name: the name of the dataset on the Hub or a local path to the dataset to train on --image_column: the name of the image column in the dataset to train on --caption_column: the name of the text column in the dataset to train on --output_dir: where to save the trained model --push_to_hub: whether to push the trained model to the Hub --checkpoi