pdufour commited on
Commit
41c8086
·
verified ·
1 Parent(s): a5dc723

Update index.js

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