Update index.js
Browse files
index.js
CHANGED
@@ -109,6 +109,12 @@ async function imageTextToText(imagePath, query, vision = true) {
|
|
109 |
);
|
110 |
input_ids.data.set(Array.from(token.data.slice(0, token.dims[1]), Number));
|
111 |
|
|
|
|
|
|
|
|
|
|
|
|
|
112 |
const dummy = new ort.Tensor("int32", new Int32Array([0]), []);
|
113 |
let { position_ids } = await ortSessionC.run({ dummy });
|
114 |
|
|
|
109 |
);
|
110 |
input_ids.data.set(Array.from(token.data.slice(0, token.dims[1]), Number));
|
111 |
|
112 |
+
|
113 |
+
let { hidden_states } = await ortSessionB.run({
|
114 |
+
input_ids: input_ids,
|
115 |
+
ids_len: ids_len,
|
116 |
+
});
|
117 |
+
|
118 |
const dummy = new ort.Tensor("int32", new Int32Array([0]), []);
|
119 |
let { position_ids } = await ortSessionC.run({ dummy });
|
120 |
|