Enum GPT2

pub enum GPT2 {
    GPT2,
    GPT2LmHead,
}

Large transformer-based language model that given a sequence of words within some text, predicts the next word.

Transformer-based language model for text generation.

Source: https://github.com/onnx/models/tree/master/text/machine_comprehension/gpt-2

Variant downloaded: ONNX Version 1.6 with Opset Version 10.

Variants

GPT2

Pure GPT2

GPT2LmHead

GPT2 + script changes

See https://github.com/onnx/models/blob/master/text/machine_comprehension/gpt-2/dependencies/GPT2-export.py for the script changes.

Trait Implementations

impl Clone for GPT2

fn clone(&self) -> GPT2

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GPT2

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl From<GPT2> for AvailableOnnxModel

fn from(model: GPT2) -> Self

Converts to this type from the input type.