Attention.__init__() got an unexpected keyword argument 'kv_heads'
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 11
- Forks
- 1
- PR merge metrics
- No merged PRs in 30d
Description
Hi,
Cloning cloned AlphaDev or pip-instally, and then running example.py returns the following error:
"TypeError: Attention.init() got an unexpected keyword argument 'kv_heads'"
I suspect the parameter attn_kv_heads in AlphaDev.init should be renamed attn_one_kv_head.
Is that correct?
Thanks for your amazing work!
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with example.py and trace the call into AlphaDev.init and Attention.init. Compare the kv_heads argument with the declared attention parameter, then confirm the example runs without the TypeError and that the parameter naming is consistent.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100