pdufour commited on
Commit
4ea7924
·
verified ·
1 Parent(s): 9656440

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +3 -0
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
  });