Fix the new stop_token_ids argument (#1558)

We forgot the default somewhere
keras-team · Apr 6, 2024 · b50f19c · b50f19c
1 parent 1c0c02a
commit b50f19c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_nlp/models/causal_lm.py b/keras_nlp/models/causal_lm.py
@@ -299,7 +299,7 @@ def generate(
         self,
         inputs,
         max_length=None,
-        stop_token_ids=None,
+        stop_token_ids="auto",
     ):
         """Generate text given prompt `inputs`.