Fix newline characters for pali_gemma (keras-team#1655)

hertschuh · May 29, 2024 · 38c6608 · 38c6608
1 parent 619a951
commit 38c6608
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -60,15 +60,15 @@ 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"]
       }
     )
 
     # Generate with batched prompts.
     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"]
       }
     )
     ```