Spaces:

lmz
/

candle-llama2

Running

App Files Files Community

lmz

radames commited on Sep 27, 2023

Commit

66ff768

•

1 Parent(s): 7c65fbc

minor ui changes (#4)

Browse files

- minor ui changes (dc843c4a87d1f22ae05772f0982743d3b8181797)
- fix first token (c0a0d79638236daaede01e3b5bc9d65d9c87ac16)

Co-authored-by: Radamés Ajna <radames@users.noreply.huggingface.co>

Files changed (2) hide show

index.html +103 -85
llama2cWorker.js +9 -2

index.html CHANGED Viewed

@@ -6,7 +6,7 @@
   <body></body>
 </html>
-<!doctype html>
 <html>
   <head>
     <meta charset="UTF-8" />
@@ -56,6 +56,7 @@
         const weightsURL = `${MODELS_BASE_URL}/${model.url}`;
         const prompt = getValue("prompt");
         const temperature = getValue("temperature");
         const repeatPenalty = getValue("repeat_penalty");
         const seed = getValue("seed");
         const maxSeqLen = getValue("max-seq");
@@ -99,6 +100,7 @@
             tokenizerURL: "tokenizer.json",
             prompt,
             temp: temperature,
             repeatPenalty,
             seed: BigInt(seed),
             maxSeqLen,
@@ -111,8 +113,18 @@
           const handleMessage = (event) => {
             const { status, error, message, prompt, sentence } = event.data;
             if (status) updateStatus(event.data);
-            if (error) reject(new Error(error));
-            if (status === "complete") resolve(event.data);
           };
           controller.signal.addEventListener("abort", handleAbort);
@@ -204,8 +216,7 @@
         <label for="model" class="font-medium">Models Options: </label>
         <select
           id="model"
-          class="border-2 border-gray-500 rounded-md font-light"
-        >
           <option value="stories15M" selected>stories 15M (60.8 MB)</option>
           <option value="stories42M">stories 42M (167 MB)</option>
           <option value="stories110M">stories 110M (438 MB)</option>
@@ -213,117 +224,124 @@
       </div>
       <form
         id="form"
-        class="flex text-normal px-1 py-1 border border-gray-700 rounded-md items-center"
-      >
         <input type="submit" hidden />
         <input
           type="text"
           id="prompt"
           class="font-light w-full px-3 py-2 mx-1 resize-none outline-none"
           placeholder="Add your prompt here..."
-          value="Once upon a time"
-        />
         <button id="clear-btn">
           <svg
             fill="none"
             xmlns="http://www.w3.org/2000/svg"
             width="40"
-            viewBox="0 0 70 40"
-          >
             <path opacity=".5" d="M39 .2v40.2" stroke="#1F2937" />
             <path
               d="M1.5 11.5 19 29.1m0-17.6L1.5 29.1"
               opacity=".5"
               stroke="#1F2937"
-              stroke-width="2"
-            />
           </svg>
         </button>
         <button
           id="run"
-          class="bg-gray-700 hover:bg-gray-800 text-white font-normal py-2 w-16 rounded disabled:bg-gray-300 disabled:cursor-not-allowed"
-        >
           Run
         </button>
       </form>
-      <div class="grid grid-cols-3 max-w-md items-center gap-3">
-        <label class="text-sm font-medium" for="max-seq">Maximum length </label>
-        <input
-          type="range"
-          id="max-seq"
-          name="temperature"
-          min="1"
-          max="256"
-          step="1"
-          value="200"
-          oninput="this.nextElementSibling.value = Number(this.value)"
-        />
-        <output
-          class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md"
-        >
-          200</output
-        >
-        <label class="text-sm font-medium" for="temperature">Temperature</label>
-        <input
-          type="range"
-          id="temperature"
-          name="temperature"
-          min="0"
-          max="2"
-          step="0.01"
-          value="0.50"
-          oninput="this.nextElementSibling.value = Number(this.value).toFixed(2)"
-        />
-        <output
-          class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md"
-        >
-          0.50</output
-        >
-        <label class="text-sm font-medium" for="repeat_penalty"
-          >Repeat Penalty</label
-        >
-        <input
-          type="range"
-          id="repeat_penalty"
-          name="repeat_penalty"
-          min="1"
-          max="2"
-          step="0.01"
-          value="1.10"
-          oninput="this.nextElementSibling.value = Number(this.value).toFixed(2)"
-        />
-        <output
-          class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md"
-          >1.10</output
-        >
-        <label class="text-sm font-medium" for="seed">Seed</label>
-        <input
-          type="number"
-          id="seed"
-          name="seed"
-          value="299792458"
-          class="font-light border border-gray-700 text-right rounded-md p-2"
-        />
-        <button
-          id="run"
-          onclick="document.querySelector('#seed').value = BigInt(Math.floor(Math.random() * 2**64-1))"
-          class="bg-gray-700 hover:bg-gray-800 text-white font-normal py-1 w-[50px] rounded disabled:bg-gray-300 disabled:cursor-not-allowed text-sm"
-        >
-          Rand
-        </button>
-      </div>
       <div>
         <h3 class="font-medium">Generation:</h3>
         <div
-          class="min-h-[250px] bg-slate-100 text-gray-500 p-4 rounded-md flex flex-col gap-2"
-        >
           <div
             id="output-counter"
             hidden
-            class="ml-auto font-semibold grid-rows-1 text-sm"
-          ></div>
           <p hidden id="output-generation" class="grid-rows-2"></p>
           <span id="output-status" class="m-auto font-light"
             >No output yet</span

   <body></body>
 </html>
+<!DOCTYPE html>
 <html>
   <head>
     <meta charset="UTF-8" />
         const weightsURL = `${MODELS_BASE_URL}/${model.url}`;
         const prompt = getValue("prompt");
         const temperature = getValue("temperature");
+        const topP = getValue("top-p");
         const repeatPenalty = getValue("repeat_penalty");
         const seed = getValue("seed");
         const maxSeqLen = getValue("max-seq");
             tokenizerURL: "tokenizer.json",
             prompt,
             temp: temperature,
+            top_p: topP,
             repeatPenalty,
             seed: BigInt(seed),
             maxSeqLen,
           const handleMessage = (event) => {
             const { status, error, message, prompt, sentence } = event.data;
             if (status) updateStatus(event.data);
+            if (error) {
+              llamaWorker.removeEventListener("message", handleMessage);
+              reject(new Error(error));
+            }
+            if (status === "aborted") {
+              llamaWorker.removeEventListener("message", handleMessage);
+              resolve(event.data);
+            }
+            if (status === "complete") {
+              llamaWorker.removeEventListener("message", handleMessage);
+              resolve(event.data);
+            }
           };
           controller.signal.addEventListener("abort", handleAbort);
         <label for="model" class="font-medium">Models Options: </label>
         <select
           id="model"
+          class="border-2 border-gray-500 rounded-md font-light">
           <option value="stories15M" selected>stories 15M (60.8 MB)</option>
           <option value="stories42M">stories 42M (167 MB)</option>
           <option value="stories110M">stories 110M (438 MB)</option>
       </div>
       <form
         id="form"
+        class="flex text-normal px-1 py-1 border border-gray-700 rounded-md items-center">
         <input type="submit" hidden />
         <input
           type="text"
           id="prompt"
           class="font-light w-full px-3 py-2 mx-1 resize-none outline-none"
           placeholder="Add your prompt here..."
+          value="Once upon a time" />
         <button id="clear-btn">
           <svg
             fill="none"
             xmlns="http://www.w3.org/2000/svg"
             width="40"
+            viewBox="0 0 70 40">
             <path opacity=".5" d="M39 .2v40.2" stroke="#1F2937" />
             <path
               d="M1.5 11.5 19 29.1m0-17.6L1.5 29.1"
               opacity=".5"
               stroke="#1F2937"
+              stroke-width="2" />
           </svg>
         </button>
         <button
           id="run"
+          class="bg-gray-700 hover:bg-gray-800 text-white font-normal py-2 w-16 rounded disabled:bg-gray-300 disabled:cursor-not-allowed">
           Run
         </button>
       </form>
+      <details>
+        <summary class="font-medium cursor-pointer">Advanced Options</summary>
+        <div class="grid grid-cols-3 max-w-md items-center gap-3 py-3">
+          <label class="text-sm font-medium" for="max-seq"
+            >Maximum length
+          </label>
+          <input
+            type="range"
+            id="max-seq"
+            name="max-seq"
+            min="1"
+            max="256"
+            step="1"
+            value="200"
+            oninput="this.nextElementSibling.value = Number(this.value)" />
+          <output
+            class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md">
+            200</output
+          >
+          <label class="text-sm font-medium" for="temperature"
+            >Temperature</label
+          >
+          <input
+            type="range"
+            id="temperature"
+            name="temperature"
+            min="0"
+            max="2"
+            step="0.01"
+            value="0.40"
+            oninput="this.nextElementSibling.value = Number(this.value).toFixed(2)" />
+          <output
+            class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md">
+            0.40</output
+          >
+          <label class="text-sm font-medium" for="top-p">Top-p</label>
+          <input
+            type="range"
+            id="top-p"
+            name="top-p"
+            min="0"
+            max="1"
+            step="0.01"
+            value="1.00"
+            oninput="this.nextElementSibling.value = Number(this.value).toFixed(2)" />
+          <output
+            class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md">
+            1.00</output
+          >
+          <label class="text-sm font-medium" for="repeat_penalty"
+            >Repeat Penalty</label
+          >
+          <input
+            type="range"
+            id="repeat_penalty"
+            name="repeat_penalty"
+            min="1"
+            max="2"
+            step="0.01"
+            value="1.10"
+            oninput="this.nextElementSibling.value = Number(this.value).toFixed(2)" />
+          <output
+            class="text-xs w-[50px] text-center font-light px-1 py-1 border border-gray-700 rounded-md"
+            >1.10</output
+          >
+          <label class="text-sm font-medium" for="seed">Seed</label>
+          <input
+            type="number"
+            id="seed"
+            name="seed"
+            value="299792458"
+            class="font-light border border-gray-700 text-right rounded-md p-2" />
+          <button
+            id="run"
+            onclick="document.querySelector('#seed').value = BigInt(Math.floor(Math.random() * 2**64-1))"
+            class="bg-gray-700 hover:bg-gray-800 text-white font-normal py-1 w-[50px] rounded disabled:bg-gray-300 disabled:cursor-not-allowed text-sm">
+            Rand
+          </button>
+        </div>
+      </details>
       <div>
         <h3 class="font-medium">Generation:</h3>
         <div
+          class="min-h-[250px] bg-slate-100 text-gray-500 p-4 rounded-md flex flex-col gap-2">
           <div
             id="output-counter"
             hidden
+            class="ml-auto font-semibold grid-rows-1 text-sm"></div>
           <p hidden id="output-generation" class="grid-rows-2"></p>
           <span id="output-status" class="m-auto font-light"
             >No output yet</span

llama2cWorker.js CHANGED Viewed

@@ -50,6 +50,7 @@ async function generate(data) {
     tokenizerURL,
     prompt,
     temp,
     repeatPenalty,
     seed,
     maxSeqLen,
@@ -59,11 +60,17 @@ async function generate(data) {
     const model = await Llama2C.getInstance(weightsURL, modelID, tokenizerURL);
     self.postMessage({ status: "loading", message: "Initializing model" });
-    model.init_with_prompt(prompt, temp, repeatPenalty, seed);
     const seq_len = model.get_seq_len();
-    let sentence = "";
     let maxTokens = maxSeqLen ? maxSeqLen : seq_len - prompt.length - 1;
     let startTime = performance.now();
     let tokensCount = 0;

     tokenizerURL,
     prompt,
     temp,
+    top_p,
     repeatPenalty,
     seed,
     maxSeqLen,
     const model = await Llama2C.getInstance(weightsURL, modelID, tokenizerURL);
     self.postMessage({ status: "loading", message: "Initializing model" });
+    const firstToken = model.init_with_prompt(
+      prompt,
+      temp,
+      top_p,
+      repeatPenalty,
+      seed
+    );
     const seq_len = model.get_seq_len();
+    let sentence = firstToken;
     let maxTokens = maxSeqLen ? maxSeqLen : seq_len - prompt.length - 1;
     let startTime = performance.now();
     let tokensCount = 0;