message: Fixes GitHub 34283
diff: @@ -266,7 +266,7 @@ class Attention(BaseDenseAttention):
  value_input = tf.keras.Input(shape=(None,), dtype='int32')
  # Embedding lookup.
  - token_embedding = tf.keras.layers.Embedding(max_tokens, dimension)
  + token_embedding = tf.keras.layers.Embedding(input_dim=1000, output_dim=64)
  # Query embeddings of shape [batch_size, Tq, dimension].
  query_embeddings = token_embedding(query_input)
  # Value embeddings of shape [batch_size, Tv, dimension].
  