chatgpt-plugin/utils/openai/tokenizer.ts
2023-06-22 18:30:08 +08:00

8 lines
No EOL
227 B
TypeScript

import { getEncoding } from 'js-tiktoken'
// TODO: make this configurable
const tokenizer = getEncoding('cl100k_base')
export function encode(input: string): Uint32Array {
return new Uint32Array(tokenizer.encode(input))
}