2

I am trying to use the Roberta-base model using AutoTokenizer.from_pretrained('roberta-base') but I get the following error:

RuntimeError: Failed to import transformers.modeling_tf_utils because of the following error (look up to see its traceback):
No module named 'tensorflow.python.keras.engine.keras_tensor'

I have tried to install tensorflow but still the same error. Any idea what is going on?

JoB
  • 23
  • 3

1 Answers1

1

You need to install with GPU, try this:

pip install --ignore-installed --upgrade tensorflow-gpu

See here for more details.

DvdG
  • 724
  • 1
  • 8
  • 15