Failed to import transformers.models.bart.modeling_tf_bart because no module named 'keras'

[jybsuper]

The following line of code causes an error: ModuleNotFoundError: No module named 'keras' whenever I try to initialize a Bart model:

transformers/src/transformers/modeling_tf_utils.py

Line 39 in 0a632f0

from keras.saving.hdf5_format import save_attributes_to_hdf5_group

Replaced it with from tensorflow.python.keras.saving.hdf5_format import save_attributes_to_hdf5_group and the error was gone.
Is this a bug or I didn't install all necessary packages?

[joshh20]

Just had this exact same issue. Thank you for the fix!

[ZJaume]

Shouldn't this be open until gets fixed?

[aymericdamien]

Confirming the issue still exists when using latest TF (2.11). Downgrading TF to 2.9 fix the issue.

[EquinetPaul]

https://stackoverflow.com/questions/74586892/no-module-named-keras-saving-hdf5-format

working for me on TF==2.9