25:["$","title",null,{"children":"GPT-3 token encoder and decoder | sj_test | Observable"}] 49:[] 29:{"id":"ef86de166ca70821","slug":null,"title":"GPT-3 token encoder and decoder","thumbnail":"579a453f9f11af734352f5ac680f48126c539728fb91f3de8f089e1e020caa9a","access_level":"public","create_time":"$D2023-05-13T06:11:55.398Z","update_time":"$D2023-05-13T06:11:55.398Z","view_time":null,"first_public_time":"$D2023-05-13T06:11:55.398Z","has_unpublished_edits":null,"fork":{"id":"5523771744f27e16","slug":"gpt-tokenizer","title":"GPT token encoder and decoder","owner":{"id":"3b974c2db4452cfc","login":"simonw","name":"Simon Willison"}},"stars":1,"starred":false,"owner":{"type":"team","id":"9f3f022b96253a72","login":"sj-test","name":"sj_test","avatar_url":"https://avatars-next.observableusercontent.com/27919a2b9e478a763acdc598f2faf4c475e544e8c9ea04fe449664f9cc56cc4d","tier":"starter_2024","bio":"","followed_by_you":false,"follows_you":false},"version":327,"publish_version":null,"role":null,"authored":false,"authors":[{"type":"individual","id":"39e0eea47bf025bf","login":"shaunjohann","name":"SJ","avatar_url":"https://avatars-next.observableusercontent.com/c9b08d00e5e1c1fc1926c76bfc4582777b45942d22b2f1b563e10fa482013637","tier":null,"bio":"","followed_by_you":false,"follows_you":false}],"members":[{"type":"individual","id":"39e0eea47bf025bf","login":"shaunjohann","name":"SJ","avatar_url":"https://avatars-next.observableusercontent.com/c9b08d00e5e1c1fc1926c76bfc4582777b45942d22b2f1b563e10fa482013637","tier":null,"role":"editor"},{"type":"team","id":"9f3f022b96253a72","login":"sj-test","name":"sj_test","avatar_url":"https://avatars-next.observableusercontent.com/27919a2b9e478a763acdc598f2faf4c475e544e8c9ea04fe449664f9cc56cc4d","tier":"starter_2024","role":"editor"}],"comments":[],"custom_thumbnail":false,"trashed":false,"lapsed":null,"subscribed":null,"forks":0,"collection":null,"type":"notebook","body":{"titling":"auto","title":"GPT-3 token encoder and decoder","cells":[{"id":72,"value":"# GPT-3 token encoder and decoder\n\nEnter text to tokenize it:","pinned":false,"hidden":false,"mode":"md","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"","database":"$undefined"},{"id":77,"value":"viewof text = Inputs.textarea({ value: \"Example text is here\" })","pinned":false,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":82,"value":"html`${encode(text).join(\" \")}`","pinned":false,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":160,"value":"html`${spans}`","pinned":false,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":88,"value":"Or convert tokens to text:","pinned":false,"hidden":false,"mode":"md","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"","database":"$undefined"},{"id":93,"value":"viewof tokens = Inputs.textarea({\n placeholder: \"Paste space separated tokens here\"\n})","pinned":false,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":106,"value":"html`

${decode(tokens.split(/[^\\d+]/g).map((t) => parseInt(t, 10)))}

`","pinned":false,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":135,"value":"## Search tokens","pinned":false,"hidden":false,"mode":"md","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"","database":"$undefined"},{"id":125,"value":"viewof matches = Inputs.search(Object.entries(decoder), {\n placeholder: \"Search for a token\"\n})","pinned":false,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":131,"value":"Inputs.table(matches.map((t) => ({ integer: t[0], token: t[1] })))","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":0,"value":"Or... paste a JSON prompt in below, e.g.\n\n```json\n{\n \"model\": \"text-davinci-003\",\n \"prompt\": [\n [\n 50256, 14832, 25, 775, 550, 257, 1049, 2652, 13, 367,\n 14132, 287, 262, 12269, 373, 28294, 290, 262, 2057,\n 318, 1107, 922, 13, 198, 48, 25, 1867, 318, 262,\n 10238, 15598, 286, 428, 2423, 290, 1521, 30, 198, 32, 25\n ]\n ],\n \"max_tokens\": 64,\n \"temperature\": 0,\n \"logprobs\": 5,\n \"user\": \"lmql\",\n \"stream\": true,\n \"echo\": true,\n \"logit_bias\": {\n \"198\": -100,\n \"628\": -100,\n \"44320\": -100\n }\n}\n```","pinned":false,"hidden":false,"mode":"md","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"","database":"$undefined"},{"id":31,"value":"viewof prompt_raw = Inputs.textarea({ rows: 10 })","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":45,"value":"html`

${\n  prompt.prompt && decode(prompt.prompt[0])\n}

Logits${\n prompt.logit_bias &&\n Object.keys(prompt.logit_bias)\n .map((k) => `
${k} - \"${decode([k])}\": ${prompt.logit_bias[k]}`)\n .join(\"\")\n}`","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":185,"value":"spans = await generateSpans(text)","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":162,"value":"async function generateSpans(text) {\n const tokens = encode(text);\n const htmlPromises = bundleTokens(tokens).map(\n async (t) =>\n `${t.tokens.join(\n \"
\"\n )}${t.text.replace(\"\\n\", \"
\").replace(\" \", \" \")}`\n );\n const html = await Promise.all(htmlPromises);\n return html.join(\"\");\n}","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":309,"value":"bundleTokens(encode(text))","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":37,"value":"prompt = {\n try {\n return JSON.parse(prompt_raw);\n } catch (e) {\n return {};\n }\n}","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":5,"value":"import {\n encode,\n byte_encoder,\n decodeStr,\n decoder\n} from \"@codingwithfire/gpt-3-encoder\"","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":9,"value":"encode(\"a and b and c\")","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":22,"value":"byte_decoder = {\n let tmp = {};\n Object.keys(byte_encoder).map((x) => {\n tmp[byte_encoder[x]] = x;\n });\n return tmp;\n}","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":25,"value":"function decode(tokens) {\n let text = tokens.map((x) => decoder[x]).join(\"\");\n text = decodeStr(text.split(\"\").map((x) => byte_decoder[x]));\n return text;\n}","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":154,"value":"decode([1276])\n","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":170,"value":"async function stringToPastelColor(str) {\n const msgUint8 = new TextEncoder().encode(str);\n const hashBuffer = await crypto.subtle.digest(\"SHA-256\", msgUint8);\n const hashArray = Array.from(new Uint8Array(hashBuffer)).slice(0, 3); // Taking only the first 3 bytes\n let color = \"#\";\n for (const value of hashArray) {\n let adjustedValue = Math.floor(value / 2) + 128; // Adjust the range to be between 128 and 256\n color += (\"00\" + adjustedValue.toString(16)).substr(-2);\n }\n return color;\n}","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":258,"value":"## Solving the problem with multi-token characters, e.g. Japanese\n\nThis tool wasn't working correctly with Japanese characters. The character 本 for example was rendered like this:\n\n � �\n\nThis is because 本 is represented as two tokens - 17312 105 - and when you revers decode those tokens individually you get back an invalid character. The following expression:\n\n decode([17312]).charCodeAt(0) // This does the same thing for 105 as well\n\nReturns 65533 - which `String.fromCharCode(65533)` confirms is the Unicode \"replacement character\" indicating an invalid character.\n\nThis is a problem because the tool likes to wrap each decoded token in a `` in order to colour highlight it.\n\nThe solution is to scan the entire response and try to find any tokens which, on their own, decode to 65533 but when combined with the subsequent token instead decode to a valid character.\n\nThese character sequences can then be bundled together into a single ``","pinned":false,"hidden":false,"mode":"md","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"","database":"$undefined"},{"id":230,"value":"[decode([17312]).charCodeAt(0), decode([103]).charCodeAt(0)]","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":249,"value":"String.fromCharCode(65533)","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":270,"value":"[decode([17312, 103]).charCodeAt(0), decode([17312, 103])]","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":277,"value":"function bundleTokens(tokens) {\n // Given an array of integer tokens, return an array of {\"string\": \"...\", \"tokens\" [1, 3]}\n let bundles = [];\n let copied = Array.from(tokens);\n let current = [];\n while (copied.length) {\n current.push(copied.shift());\n if (decode(current).charCodeAt(0) == 65533) {\n continue;\n }\n bundles.push({ text: decode(current), tokens: current });\n current = [];\n }\n return bundles;\n}","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":324,"value":"Here's a bug I still need to resolve - the string \"本本\" breaks things, as seen below:","pinned":false,"hidden":false,"mode":"md","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"","database":"$undefined"},{"id":326,"value":"encode(\"本本\")","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"},{"id":297,"value":"JSON.stringify(bundleTokens([17312, 105, 42164, 105]))","pinned":true,"hidden":false,"mode":"ojs","author":"$undefined","summary":"$undefined","tool":"$undefined","providerMetadata":"$undefined","attachments":"$undefined","output":"$undefined","database":"$undefined"}],"files":"$W49","stdlib":"1"}} 24:null