chunked/nltk_chunking/model_doc_gpt_bigcode/chunk_21.txt

- Use the memory layout (self.num_heads, 3, self.head_dim) instead of (3, self.num_heads, self.head_dim) for the QKV tensor with MHA.