Update index.js
Browse files
index.js
CHANGED
|
@@ -207,11 +207,13 @@ export async function imageTextToText(
|
|
| 207 |
|
| 208 |
const dummy = new ort.Tensor("int32", new Int32Array([0]), []);
|
| 209 |
|
|
|
|
| 210 |
let { hidden_states } = await ortSessionB.run({
|
| 211 |
input_ids: input_ids,
|
| 212 |
ids_len: ids_len,
|
| 213 |
});
|
| 214 |
|
|
|
|
| 215 |
({ position_ids } = await ortSessionC.run({
|
| 216 |
dummy: dummy,
|
| 217 |
}));
|
|
@@ -229,6 +231,7 @@ export async function imageTextToText(
|
|
| 229 |
image = image.div_(255.0);
|
| 230 |
const pixel_values = image.unsqueeze(0);
|
| 231 |
|
|
|
|
| 232 |
const { image_embed } = await ortSessionA.run({
|
| 233 |
pixel_values: pixel_values,
|
| 234 |
});
|
|
|
|
| 207 |
|
| 208 |
const dummy = new ort.Tensor("int32", new Int32Array([0]), []);
|
| 209 |
|
| 210 |
+
console.log("RUN SESSION B");
|
| 211 |
let { hidden_states } = await ortSessionB.run({
|
| 212 |
input_ids: input_ids,
|
| 213 |
ids_len: ids_len,
|
| 214 |
});
|
| 215 |
|
| 216 |
+
console.log("RUN SESSION C");
|
| 217 |
({ position_ids } = await ortSessionC.run({
|
| 218 |
dummy: dummy,
|
| 219 |
}));
|
|
|
|
| 231 |
image = image.div_(255.0);
|
| 232 |
const pixel_values = image.unsqueeze(0);
|
| 233 |
|
| 234 |
+
console.log("RUN SESSION A");
|
| 235 |
const { image_embed } = await ortSessionA.run({
|
| 236 |
pixel_values: pixel_values,
|
| 237 |
});
|