Remove endoftext from the end of the prompt in the example.

SamanehSaadat · May 23, 2024 · 41b084a · 41b084a
1 parent 316e122
commit 41b084a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/keras_nlp/src/models/falcon/falcon_causal_lm.py b/keras_nlp/src/models/falcon/falcon_causal_lm.py
@@ -73,10 +73,10 @@ class FalconCausalLM(CausalLM):
     Use `generate()` without preprocessing.
     ```python
     prompt = {
-        # Token ids for "<|endoftext|> Keras is<|endoftext|>".
-        "token_ids": np.array([[50256, 17337,   292,   318, 50256, 0]] * 2),
+        # Token ids for "<|endoftext|> Keras is".
+        "token_ids": np.array([[50256, 17337,   292,   318]] * 2),
         # Use `"padding_mask"` to indicate values that should not be overridden.
-        "padding_mask": np.array([[1, 1, 1, 1, 1, 0]] * 2),
+        "padding_mask": np.array([[1, 1, 1, 1]] * 2),
     }
 
     falcon_lm = keras_nlp.models.FalconCausalLM.from_preset(