examples/research_projects/vqgan-clip/VQGAN_CLIP.py/ProcessorGradientFlow/__call__
class ProcessorGradientFlow:    def __call__(self, text=None, images=None, **kwargs):
        encoding = self.tokenizer(text=text, **kwargs)
        encoding["pixel_values"] = self.preprocess_img(images)
        encoding = {key: value.to(self.device) for (key, value) in encoding.items()}
        return encoding