google-research / bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.
https://arxiv.org/abs/2004.04696
Apache License 2.0
697 stars 85 forks source link

What tensorflow versions are possible? #59

Closed lambdaofgod closed 6 months ago

lambdaofgod commented 11 months ago

I am trying to run bleurt with tensorflow 2.15 and I get

TypeError: Binding inputs to tf.function failed due to `got an unexpected keyword argument 'input_ids'`. Received args: () and kwargs: {'input_ids': <tf.Tensor: shape=(10, 128), dtype=int64, numpy=

Which looks like my tensorflow is too new.

What is the newest tensorflow version supported by bleurt?

deepvine commented 11 months ago

In my case, it worked with version 2.13.1