@syonfox/gpt-3-encoder
v1.4.0-rc5
Published
Javascript BPE Encoder Decoder for GPT-2 / GPT-3. The "gpt-3-encoder" module provides functions for encoding and decoding text using the Byte Pair Encoding (BPE) algorithm. It can be used to process text data for input into machine learning models, or to
Downloads
446
Maintainers
Readme
GPT-3-Encoder
Javascript library for encoding and decoding text using Byte Pair Encoding (BPE), as used in GPT-2 and GPT-3 models by OpenAI. This is a fork of the original python implementation by OpenAI, which can be found here.
This fork includes additional features such as the countTokens and tokenStats functions, as well as updated documentation.
Installation
To install with npm:
npm install gpt-3-encoder
Usage
Also check out the browser demo browser demo
Compatible with Node >= 12
To use the library in your project, import it as follows:
const GPT3Encoder = require('gpt-3-encoder');
Additional Features
In addition to the original encoding
and decoding
functions, this fork includes the following additional features:
countTokens(text: string): number
This function returns the number of tokens in the provided text, after encoding it using BPE.
tokenStats(text: string): object
This function returns an object containing statistics about the tokens in the provided text, after encoding it using BPE. The returned object includes the following properties:
count
: the total number of tokens in the text.unique
: the number of unique tokens in the text.frequencies
: an object containing the frequency of each token in the text.postions
: an object mapping tokens to positions in the encoded stringtokens
: same as the output to tokens Compatibility
This library is compatible with both Node.js and browser environments, we have used webpack to build /dist/bundle.js 1.5 MB including the data. A compiled version for both environments is included in the package. Credits
This library was created as a fork of the original GPT-3-Encoder library by latitudegames.
Example
See browser.html and demo.js Note you may need to include it from the appropriate place in node modules / npm package name
import {encode, decode, countTokens, tokenStats} from "gpt-3-encoder"
//or note you might need @syonfox/gpt-3-encoder if thats what you npm install
const {encode, decode, countTokens, tokenStats} = require('gpt-3-encoder')
const str = 'This is an example sentence to try encoding out on!'
const encoded = encode(str)
console.log('Encoded this string looks like: ', encoded)
console.log('We can look at each token and what it represents')
for (let token of encoded) {
console.log({token, string: decode([token])})
}
//example count tokens usage
if (countTokens(str) > 5) {
console.log("String is over five tokens, inconcevable");
}
const decoded = decode(encoded)
console.log('We can decode it back into:\n', decoded)
Developers
I have added som other examples to the examples folder. Please take a look at package.json for how to do stuff
git clone https://github.com/syonfox/GPT-3-Encoder.git
cd GPT-3-Encoder
npm install # install dev deps (docs tests build)
npm run test # run tests
npm run docs # build docs
npm run build # builds it for the browser
npm run browser # launches demo inf firefox
npm run demo # runs node.js demo
less Encoder.js # the main code is here
firefox ./docs/index.html # view docs locally
npm publish --access public # dev publish to npm
Performance
Built bpe_ranks in 100 ms
// using js loading (probably before cache) Loaded encoder in 121 ms Loaded bpe_ranks in 91 ms
// using fs loading Loaded encoder in 32 ms Loaded bpe_ranks in 44 ms
//back to js loading Loaded encoder in 35 ms Loaded bpe_ranks in 40 ms
todo
More stats that work well with this token representation.
Clean up and keep it simple.
Here are some additional suggestions for improving the GPT-3 Encoder:
- Add more unit tests to ensure the correctness and reliability of the code. This can be particularly important for the encode and decode functions, which are the main functions of the encoder.
- Add more documentation and examples to help users understand how to use the encoder and integrate it into their own projects. This could include additional JSDoc comments, as well as additional documentation in the README file and/or GitHub Pages.
- Consider adding support for other languages and character sets. Currently, the encoder only supports ASCII characters, but there may be a demand for support for other languages and character sets.
- Explore potential optimizations and performance improvements for the encode and decode functions. Some ideas might include using faster data structures (such as a hash map or a trie), implementing more efficient algorithms, or using multi-threading or web workers to take advantage of multiple cores or processors.
- Consider adding support for other models or use cases. For example, you could add support for other OpenAI models ( such as GPT-2 or GPT-3) or for other applications of BPE encoding (such as machine translation or natural language processing).