older="unet")
model.register_to_config(**load_model.config)

model.load_state_dict(load_model.state_dict()) Next, the text and image columns of the dataset need to be preprocessed. The tokenize_captions function handles tokenizing the inputs, and the train_transforms function specifies the type of transforms to apply to the image. Both of these functions are bundled into preprocess_train:   Copied def preprocess_train(examples):
    images = [i