What does this implement/fix? Explain your changes.
Fixes previous build failure in main branch.
Does your contribution introduce a new dependency? If yes, which one?
Enforces keras version 2.5.0.rc0 and tensorflow 2.5.0. Both need to be these versions to make the keras-self-attention package work properly. Everything else should work fine with different versions of keras/tensorflow.
Reference Issues/PRs
What does this implement/fix? Explain your changes.
Fixes previous build failure in main branch.
Does your contribution introduce a new dependency? If yes, which one?
Enforces keras version 2.5.0.rc0 and tensorflow 2.5.0. Both need to be these versions to make the keras-self-attention package work properly. Everything else should work fine with different versions of keras/tensorflow.
Any other comments?