Skip to content

Commit 1549b51

Browse files
Use keras.backend.floatx() instead of keras.config.floatx()
for Keras 2 compatibility
1 parent 45b03a5 commit 1549b51

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_nlp/models/mistral/mistral_backbone.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(
4545
**kwargs,
4646
):
4747
# Get the dtype
48-
dtype = kwargs.pop("dtype", keras.config.floatx())
48+
dtype = kwargs.pop("dtype", keras.backend.floatx())
4949

5050
# Inputs
5151
token_ids = keras.Input(shape=(None,), dtype="int32", name="token_ids")

0 commit comments

Comments
 (0)