def convert_to_fp32(self):
    self.text_encoder.apply(convert_module_to_f32)
    self.image_encoder.apply(convert_module_to_f32)
def convert_to_fp32(self):
    self.text_encoder.apply(convert_module_to_f32)
    self.image_encoder.apply(convert_module_to_f32)