GPT-2 in the browser with TensorFlow.js

This repository contains a demo of loading a GPT-2 model in the browser with TensorFlow.js. The model weights are loaded from Hugging Face, converted in real-time and then cached in IndexedDB.

Libraries used:

• tfjs - to run the model
• gpt-tfjs - to define the model architecture
• h5web - to load .h5 files dynamically
• gpt-tokenizer - to encode/decode text
• stats.js - to show stats