From b50f19ccb1525c95b12218345d9d6b07c7f87c3f Mon Sep 17 00:00:00 2001
From: Matt Watson <1389937+mattdangerw@users.noreply.github.com>
Date: Fri, 5 Apr 2024 18:40:16 -0700
Subject: [PATCH] Fix the new stop_token_ids argument (#1558)

We forgot the default somewhere
---
 keras_nlp/models/causal_lm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/keras_nlp/models/causal_lm.py b/keras_nlp/models/causal_lm.py
index 98c073b165..8595ef1fd8 100644
--- 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`.