Fix typo in position embedding docstring (#86)

keras-team · Apr 5, 2022 · 61ecffb · 61ecffb
1 parent 0fd7a89
commit 61ecffb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_nlp/layers/position_embedding.py b/keras_nlp/layers/position_embedding.py
@@ -46,8 +46,8 @@ class PositionEmbedding(keras.layers.Layer):
         max_length=max_length
     )
 
-    embedded_tokens = self.token_embeddings(inputs)
-    embedded_positions = self.position_embeddings(embedded_tokens)
+    embedded_tokens = token_embeddings(inputs)
+    embedded_positions = position_embeddings(embedded_tokens)
     outputs = embedded_tokens + embedded_positions
     ```