Skip to content

perf(minimax-h3): unpack QKV with unbind - #1698

Open
iMountTai wants to merge 5 commits into
modelscope:mainfrom
HYGON-AI:perf/minimax-h3-qkv-unbind
Open

iMountTai wants to merge 5 commits into
modelscope:mainfrom
HYGON-AI:perf/minimax-h3-qkv-unbind

Conversation

@iMountTai

Copy link
Copy Markdown

MiniMax-H3 currently extracts Q, K and V through three separate indexing expressions. Replace them with a single unbind in both attention layouts: dimension 2 for the standard model and dimension 1 for the Comfy model. This removes the separate indexing chains while preserving the Q/K/V views and downstream attention computation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant