Update src/models.js - Use conditional operator

Co-authored-by: Joshua Lochner <[email protected]>
huggingface · Nov 19, 2023 · 8bd4ccc · 8bd4ccc
1 parent f89a296
commit 8bd4ccc
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/models.js b/src/models.js
@@ -158,13 +158,10 @@ async function validateInputs(session, inputs) {
         if (!tensor) {
             missingInputs.push(inputName);
         } else {
-            if (env.wasm.proxy) {
-                // Moving the tensor across Worker boundary moves ownership to the worker,
-                // which invalidates the tensor. So we simply sacrifize the clone for it.
-                checkedInputs[inputName] = tensor.clone();
-            } else {
-                checkedInputs[inputName] = tensor;
-            }
+            // NOTE: When `env.wasm.proxy is true`, when the tensor is moved across the Worker
+            // boundary, the ownership is transferred to the worker, invalidating the tensor.
+            // So, in this case, we simply sacrifice a clone for it.
+            checkedInputs[inputName] = env.wasm.proxy ? tensor.clone() : tensor;
         }
     }
     if (missingInputs.length > 0) {