def convert_to_fp16(self):
    self.text_encoder.apply(convert_module_to_f16_2)
    self.image_encoder.apply(convert_module_to_f16)
def convert_to_fp16(self):
    self.text_encoder.apply(convert_module_to_f16_2)
    self.image_encoder.apply(convert_module_to_f16)