Check if key input is null

microsoft · Jan 6, 2025 · ef489ea · ef489ea
1 parent 0db515f
commit ef489ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts b/js/web/lib/wasm/jsep/webgpu/ops/group-query-attention.ts
@@ -87,7 +87,7 @@ export const validateInputs = (
   let kvSequenceLength = sequenceLength;
 
   let pastSequenceLength = 0;
-  const packedQKV = !key || key.dims.length === 0 || ShapeUtil.size(key.dims) === 0;
+  const packedQKV = !key || key === null ||  key.dims.length === 0 || ShapeUtil.size(key.dims) === 0;
   const headSize = !packedQKV
     ? Math.floor(hiddenSize / attributes.numHeads)
     : Math.floor(hiddenSize / (attributes.numHeads + 2 * attributes.kvNumHeads));