Update index.js

index.js

@@ -46,6 +46,21 @@ async function initializeSessions() {
     { executionProviders: ["webgpu"] }
   );
 
+  ortSessionD = await ort.InferenceSession.create(
+    await getModelFile(ONNX_MODEL, `onnx/QwenVL_D_${QUANT}.onnx`),
+    {
+      executionProviders: ["webgpu"],
+    }
+
+    ortSessionE = await ort.InferenceSession.create(
+      await getModelFile(ONNX_MODEL, `onnx/QwenVL_E_${QUANT}.onnx`),
+      {
+        executionProviders: ["wasm"],
+      },
+    );
+  );
+
+
   config = (await getModelJSON(BASE_MODEL, "config.json"));
 
   status.textContent = 'Ready';
@@ -235,14 +250,7 @@ export async function imageTextToText(
 
     await ortSessionA.release();
     ortSessionA = null;
-
-    ortSessionD = await ort.InferenceSession.create(
-      await getModelFile(ONNX_MODEL, `onnx/QwenVL_D_${QUANT}.onnx`),
-      {
-        executionProviders: ["webgpu"],
-      }
-    );
-
+    
     ({ hidden_states, position_ids } = await ortSessionD.run({
       "hidden_states.1": hidden_states,
       image_embed,
@@ -263,15 +271,6 @@ export async function imageTextToText(
   ) {
     let token_id;
 
-    if (!ortSessionE) {
-      ortSessionE = await ort.InferenceSession.create(
-        await getModelFile(ONNX_MODEL, `onnx/QwenVL_E_${QUANT}.onnx`),
-        {
-          executionProviders: ["wasm"],
-        },
-      );
-    }
-
     ({
       max_logit_ids: token_id,
       past_key_states: past_key_states,