def METHOD_NAME(self, save_dir: Union[str, os.PathLike]):
    self.qry_model.encoder.save_pretrained(os.path.join(save_dir, 'qry_encoder'))
    self.ctx_model.encoder.save_pretrained(os.path.join(save_dir, 'ctx_encoder'))