type: LitVQVAEModel
args:
  mapping: sentence_piece