From e59391924e36f4b937e20e8799cbae14e55ce2ff Mon Sep 17 00:00:00 2001 From: Matt Date: Wed, 7 Jun 2023 12:16:13 +0100 Subject: [PATCH] Correct imports for Keras 2.13 --- src/transformers/modeling_tf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py index dcf226b80b3634..5127e4c9821b28 100644 --- 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