Skip to content

Commit 38d9543

Browse files
authored
Fix ReversibleEmbedding failing with .summary() (#22434)
* fix(embedding): rename compute_output_shape param to match build signature * fix(embedding): rename build param to input_shape instead of inputs_shape
1 parent 2fb46fd commit 38d9543

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

keras/src/layers/core/reversible_embedding.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,8 @@ def __init__(
106106
self.reverse_dtype = reverse_dtype
107107
self.logit_soft_cap = logit_soft_cap
108108

109-
def build(self, inputs_shape=None):
110-
super().build(inputs_shape)
109+
def build(self, input_shape=None):
110+
super().build(input_shape)
111111
if not self.tie_weights and self.quantization_mode not in (
112112
"int8",
113113
"int4",

0 commit comments

Comments
 (0)