diff --git a/keras_nlp/src/models/pali_gemma/pali_gemma_causal_lm.py b/keras_nlp/src/models/pali_gemma/pali_gemma_causal_lm.py index b1a62d7eb7..6d2cbdf7c7 100644 --- a/keras_nlp/src/models/pali_gemma/pali_gemma_causal_lm.py +++ b/keras_nlp/src/models/pali_gemma/pali_gemma_causal_lm.py @@ -60,7 +60,7 @@ class PaliGemmaCausalLM(CausalLM): pali_gemma_lm.generate( { "images": image, - "text": ["answer en where is the cow standing?\n"] + "text": ["answer en where is the cow standing?\\n"] } ) @@ -68,7 +68,7 @@ class PaliGemmaCausalLM(CausalLM): pali_gemma_lm.generate( { "images": [image, image], - "text": ["answer en where is the cow standing?\n", "caption en\n"] + "text": ["answer en where is the cow standing?\\n", "caption en\\n"] } ) ```