Spaces:
Running
Running
force cache with cache API and fix max seq (#2)
Browse files- force cache with cache API and fix max seq (c2a511e0225b51fa9cb3c772501b002126ece76a)
Co-authored-by: Radamés Ajna <radames@users.noreply.huggingface.co>
- index.html +10 -2
- llama2cWorker.js +10 -6
index.html
CHANGED
@@ -38,11 +38,11 @@
|
|
38 |
},
|
39 |
stories42M: {
|
40 |
url: "stories42M.bin",
|
41 |
-
seq_len:
|
42 |
},
|
43 |
stories110M: {
|
44 |
url: "stories110M.bin",
|
45 |
-
seq_len:
|
46 |
},
|
47 |
};
|
48 |
|
@@ -124,9 +124,17 @@
|
|
124 |
const prompt = document.querySelector("#prompt");
|
125 |
const clearBtn = document.querySelector("#clear-btn");
|
126 |
const runBtn = document.querySelector("#run");
|
|
|
127 |
let runController = new AbortController();
|
128 |
let isRunning = false;
|
129 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
130 |
form.addEventListener("submit", async (e) => {
|
131 |
e.preventDefault();
|
132 |
if (isRunning) {
|
|
|
38 |
},
|
39 |
stories42M: {
|
40 |
url: "stories42M.bin",
|
41 |
+
seq_len: 1024,
|
42 |
},
|
43 |
stories110M: {
|
44 |
url: "stories110M.bin",
|
45 |
+
seq_len: 1024,
|
46 |
},
|
47 |
};
|
48 |
|
|
|
124 |
const prompt = document.querySelector("#prompt");
|
125 |
const clearBtn = document.querySelector("#clear-btn");
|
126 |
const runBtn = document.querySelector("#run");
|
127 |
+
const modelSelect = document.querySelector("#model");
|
128 |
let runController = new AbortController();
|
129 |
let isRunning = false;
|
130 |
|
131 |
+
modelSelect.addEventListener("change", (e) => {
|
132 |
+
const model = MODELS[e.target.value];
|
133 |
+
document.querySelector("#max-seq").max = model.seq_len;
|
134 |
+
document.querySelector("#max-seq").nextElementSibling.value =
|
135 |
+
model.seq_len;
|
136 |
+
});
|
137 |
+
|
138 |
form.addEventListener("submit", async (e) => {
|
139 |
e.preventDefault();
|
140 |
if (isRunning) {
|
llama2cWorker.js
CHANGED
@@ -1,13 +1,17 @@
|
|
1 |
import init, { Model } from "./build/m.js";
|
2 |
|
3 |
async function fetchArrayBuffer(url) {
|
4 |
-
const
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
|
|
|
|
|
|
|
|
|
|
9 |
}
|
10 |
-
|
11 |
class Llama2C {
|
12 |
static instance = {};
|
13 |
|
|
|
1 |
import init, { Model } from "./build/m.js";
|
2 |
|
3 |
async function fetchArrayBuffer(url) {
|
4 |
+
const cacheName = "llama2c-candle-cache";
|
5 |
+
const cache = await caches.open(cacheName);
|
6 |
+
const cachedResponse = await cache.match(url);
|
7 |
+
if (cachedResponse) {
|
8 |
+
const data = await cachedResponse.arrayBuffer();
|
9 |
+
return new Uint8Array(data);
|
10 |
+
}
|
11 |
+
const res = await fetch(url, { cache: "force-cache" });
|
12 |
+
cache.put(url, res.clone());
|
13 |
+
return new Uint8Array(await res.arrayBuffer());
|
14 |
}
|
|
|
15 |
class Llama2C {
|
16 |
static instance = {};
|
17 |
|