DonutModel constructor

DonutModel(
  1. DonutConfig config
)

Creates a new Donut model with the given configuration.

The model is created with random weights. Call loadWeights or fromPretrained to load pretrained weights before inference.

Implementation

DonutModel(this.config) {
  encoder = SwinEncoder(
    inputSize: config.inputSize,
    patchSize: config.patchSize,
    embedDim: config.encoderEmbedDim,
    encoderLayer: config.encoderLayer,
    numHeads: config.encoderNumHeads,
    windowSize: config.windowSize,
  );

  decoder = BartDecoder(
    decoderLayers: config.decoderLayer,
    maxPositionEmbeddings: config.maxPositionEmbeddings,
    vocabSize: config.vocabSize,
    embedDim: config.decoderEmbedDim,
    ffnDim: config.decoderFfnDim,
    numHeads: config.decoderNumHeads,
  );
}