Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
satyajandhyala committed Oct 25, 2024
1 parent 2dff8d5 commit 557cd29
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion js/web/lib/wasm/jsep/webgpu/ops/attention.ts
Original file line number Diff line number Diff line change
Expand Up @@ -856,7 +856,10 @@ export const applyAttention = (
seqLens,
totalSequenceLengthInput,
),
{ inputs: seqLens && totalSequenceLengthInput ? [probs, seqLens, totalSequenceLengthInput] : [probs], outputs: [-1] },
{
inputs: seqLens && totalSequenceLengthInput ? [probs, seqLens, totalSequenceLengthInput] : [probs],
outputs: [-1],
},
)[0];

// Run AttentionScore
Expand Down

0 comments on commit 557cd29

Please sign in to comment.