"PyTorch">
Let’s train on the Pokémon BLIP captions dataset to generate your own Pokémon. Set the environment variables MODEL_NAME and dataset_name to the model and the dataset (either from the Hub or a local path). If you’re training on more than one GPU, add the --multi_gpu parameter to the accelerate launch command. To train on a local dataset, set the TRAIN_DIR and OUTPUT_DIR environment variables to the path of the dataset and where to sav