PaliGemma docstring fix (#1651)

Missing quote leads to a rendering error on keras.io
keras-team · May 24, 2024 · 619a951 · 619a951
1 parent 12e3e99
commit 619a951
Showing 1 changed file with 4 additions and 3 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
@@ -55,7 +55,8 @@ class PaliGemmaCausalLM(CausalLM):
     ```python
     image = np.random.rand(224, 224, 3)
     pali_gemma_lm = keras_nlp.models.PaliGemmaCausalLM.from_preset(
-        "pali_gemma_pt_224")
+        "pali_gemma_3b_mix_224"
+    )
     pali_gemma_lm.generate(
       {
         "images": image,
@@ -67,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"]
       }
     )
     ```
@@ -84,7 +85,7 @@ class PaliGemmaCausalLM(CausalLM):
     }
 
     pali_gemma_lm = keras_nlp.models.PaliGemmaCausalLM.from_preset(
-        "pali_gemma_pt_224",
+        "pali_gemma_3b_mix_224",
         preprocessor=None,
     )
     pali_gemma_lm.generate(inputs)