Module model

Module model 

Source
Expand description

Runtime wrappers for running inference on a GPT model saved in ONNX format.

Structs§

Gpt2Model
The GPT-2 natural langage ML model.

Constants§

GPT2_EMBEDDING_SIZE
Number of embeddings used by each layer of the GPT-2 models supported by this library.