Correct imports for Keras 2.13

huggingface · Jun 7, 2023 · e593919 · e593919
1 parent a8d2f4c
commit e593919
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py
@@ -74,7 +74,7 @@
 if parse(tf.__version__).minor >= 13:
     from keras import backend as K
     from keras.__internal__ import KerasTensor
-    from keras.engine.base_layer_utils import call_context
+    from keras.src.engine.base_layer_utils import call_context
 elif parse(tf.__version__).minor >= 11:
     from keras import backend as K
     from keras.engine.base_layer_utils import call_context