LlamaAttention forward function type hint is incorrect from new Branch

[ArkVex]

Hi, this PR fixes a small issue in the LlamaAttention class. The return type in the forward method currently shows three values, but the function actually returns only two. This seems to have been missed during the attention refactor (possibly in PR #35235).

I’ve updated the type hint to reflect the actual return values, just to avoid confusion for anyone reading or using the code. Let me know if any other changes are needed. Happy to help!

[ArkVex]

@Rocketknight1 same error again

[ArkVex]

its saying this change is causing the issue...but this was the whole point for this pr

[yaswanth19]

Hey @ArkVex , There are many models which inherit from llama modules. So, running make fix-copies should make the CI happy

[ArkVex]

I did...but I don't know why it is throwing error

[yaswanth19]

It's working fine for me. Can't you see any changes in your branch after running. Might be that the changes have been applied and in the next run there is no diff. Otherwise install pip install -e '.[quality]' and then once again try it.

[Rocketknight1]

Hi @ArkVex, the problem is that make fix-copies detects changes by comparing against your fork's main branch and not Hugging Face's main branch. Because you've made the same changes on main, it doesn't think any files have changed in your PR. You should reset your main branch to match huggingface/transformers@main, and then pull the changes to your local machine's main branch. After that, everything should work.

[kkirsche]

If it helps, the reset command(s) may look something like:

git switch main
git remote add upstream https://github.com/huggingface/transformers.git
git reset --hard upstream/main

(X-Ref to link this to the last MR: #38795)