Skip to content

Commit c5679d8

Browse files
Merge branch 'mistral-backbone' of github.com:tirthasheshpatel/keras-nlp into mistral-backbone
2 parents ad2ae64 + 1549b51 commit c5679d8

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
@@ -110,7 +110,7 @@ def __init__(
110110
**kwargs,
111111
):
112112
# Get the dtype
113-
dtype = kwargs.pop("dtype", keras.config.floatx())
113+
dtype = kwargs.pop("dtype", keras.backend.floatx())
114114

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

0 commit comments

Comments
 (0)