PaliGemma docstring fix (#1651)

mattdangerw · web-flow · commit 619a951adc4e · 2024-05-23T17:05:15.000-07:00
Missing quote leads to a rendering error on keras.io
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)

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,8 @@ class PaliGemmaCausalLM(CausalLM):`
`55`	`55`	```python
`56`	`56`	`image = np.random.rand(224, 224, 3)`
`57`	`57`	`pali_gemma_lm = keras_nlp.models.PaliGemmaCausalLM.from_preset(`
`58`		`- "pali_gemma_pt_224")`
	`58`	`+ "pali_gemma_3b_mix_224"`
	`59`	`+ )`
`59`	`60`	`pali_gemma_lm.generate(`
`60`	`61`	`{`
`61`	`62`	`"images": image,`
`@@ -67,7 +68,7 @@ class PaliGemmaCausalLM(CausalLM):`
`67`	`68`	`pali_gemma_lm.generate(`
`68`	`69`	`{`
`69`	`70`	`"images": [image, image],`
`70`		`- "text": ["answer en where is the cow standing?\n", "caption en\n]`
	`71`	`+ "text": ["answer en where is the cow standing?\n", "caption en\n"]`
`71`	`72`	`}`
`72`	`73`	`)`
`73`	`74`	```
`@@ -84,7 +85,7 @@ class PaliGemmaCausalLM(CausalLM):`
`84`	`85`	`}`
`85`	`86`
`86`	`87`	`pali_gemma_lm = keras_nlp.models.PaliGemmaCausalLM.from_preset(`
`87`		`- "pali_gemma_pt_224",`
	`88`	`+ "pali_gemma_3b_mix_224",`
`88`	`89`	`preprocessor=None,`
`89`	`90`	`)`
`90`	`91`	`pali_gemma_lm.generate(inputs)`