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!
Pay now to fund the work behind this issue.
Get updates on progress being made.
Maintainer is rewarded once the issue is completed.
You're funding impactful open source efforts
You want to contribute to this effort
You want to get funding like this too