trocr

Model type

trocr

Class

TrOCRForConditionalGeneration

Task

seq2seq

Source

models/trocr.py

Description

TrOCR decoder model.

Architecturally identical to BART; only weight renaming differs. The vision encoder is a separate model (e.g. ViT, DeiT, BEiT).

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")