Update README.md

README.md

@@ -83,6 +83,12 @@ function collectAttentions(out: Record<string, Tensor>): XtTensor[] {
   return keys.map((k) => out[k] as unknown as XtTensor);
 }
 
+function onesMask(n: number): Tensor {
+  const data = BigInt64Array.from({ length: n }, () => 1n);
+  return new Tensor('int64', data, [1, n]);
+}
+
+
 /**
  * Tokenization:
  * Prefer the public callable form `tokenizer(text, {...})` which returns tensors.
@@ -112,7 +118,7 @@ const input_ids = Array.from(
  */
 const out = (await model.forward({
   input_ids,
-  attention_mask: windowAttentionMask,
+  attention_mask: onesMask(input_ids.length),
   output_attentions: true,
 })) as unknown as Record<string, Tensor>;