diff --git a/keras_hub/src/layers/modeling/reversible_embedding.py b/keras_hub/src/layers/modeling/reversible_embedding.py index df7ff35735..6d182ae239 100644 --- a/keras_hub/src/layers/modeling/reversible_embedding.py +++ b/keras_hub/src/layers/modeling/reversible_embedding.py @@ -115,7 +115,12 @@ def build(self, inputs_shape=None): def call(self, inputs, reverse=False): if reverse: if self.tie_weights: - kernel = ops.transpose(ops.convert_to_tensor(self.embeddings)) + # Ensure embeddings is properly converted to a tensor + embeddings_tensor = self.embeddings + # If it's a Keras variable, get its value + if hasattr(embeddings_tensor, "value"): + embeddings_tensor = embeddings_tensor.value + kernel = ops.transpose(ops.convert_to_tensor(embeddings_tensor)) else: kernel = self.reverse_embeddings if self.reverse_dtype is not None: