minor bug

microsoft · Jan 29, 2025 · 49b765e · 49b765e
1 parent 4edf220
commit 49b765e
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -379,8 +379,8 @@ export const groupQueryAttention = (context: ComputeContext, attributes: GroupQu
 
   applyAttention(
     context,
-    attributes.doRotary ? qRotary! : Q,
-    attributes.doRotary ? kRotary! : K,
+    Q,
+    K,
     V,
     undefined,
     undefined,