Update index.html

index.html

@@ -286,8 +286,8 @@
       if (mySeq !== loadSeq) return;
 
       setStatus("Warming up…");
-      const enc = await tokenizer(" ", { add_special_tokens:false });
-      await model(enc.input_ids, { attention_mask: enc.attention_mask });
+      const enc = await tokenizer(" ", { add_special_tokens: false, return_attention_mask: true });
+      await model({ input_ids: enc.input_ids, attention_mask: enc.attention_mask });
       if (mySeq !== loadSeq) return;
       setStatus("Ready");
     }
@@ -295,9 +295,9 @@
     /* ---------- Next-token logic ---------- */
     async function greedyNext(text, topK=10){
       if (!tokenizer || !model) { setErr("Model not loaded yet — check the status bar."); return {rows:[],dt:0}; }
-      const enc = await tokenizer(text || " ", { add_special_tokens:false });
+      const enc = await tokenizer(text || " ", { add_special_tokens: false, return_attention_mask: true });
       const t0 = performance.now();
-      const out = await model(enc.input_ids, { attention_mask: enc.attention_mask });
+      const out = await model({ input_ids: enc.input_ids, attention_mask: enc.attention_mask });
       const dt = (performance.now() - t0) | 0;
 
       const last = out.logits[out.logits.length - 1];