The HuggingFace transformers library already has support for a similar method called prompt lookup decoding that uses the existing context to generate an ngram

The HuggingFace transformers library already has support for a similar method called prompt lookup decoding that uses the existing context to generate an ngram model: https://github.com/huggingface/transformers/issues/27722 https://github.com/huggingface/transformers/issues/27722 I don't think it would be that hard to switch it out for a pretrained ngram model.