I was facing the exact issues and wasn’t able to find any solution online. However, the problem seems to be resolved by downgrading to tensorflow-addons==0.7.1 (and also some older versions… this is the last one working for me). After doing that I get a warning based on OpenNMT-tf requirements, so to be safe I had to downgrade to OpenNMT-tf==2.6.0.
I hope the issue will be resolved later on so I can use the latest version, but at least I can work with this one for now…