Updated docs

f-prime · f-prime · commit 78544fd7fc34 · 2023-04-17T04:34:22.000Z
diff --git a/pgml-docs/docs/user_guides/transformers/pre_trained_models.md b/pgml-docs/docs/user_guides/transformers/pre_trained_models.md
@@ -14,7 +14,7 @@ pgml.transform(
     task TEXT OR JSONB,       -- task name or full pipeline initializer arguments
     call JSONB,               -- additional call arguments alongside the inputs
     inputs TEXT[] OR BYTEA[], -- inputs for inference
-    cache_model BOOLEAN       -- if true, the model will be cached in memory. FALSE by default
+    cache BOOLEAN             -- if TRUE, the model will be cached in memory. FALSE by default.
 )
 ```
 
@@ -75,7 +75,7 @@ Sentiment analysis is one use of `text-classification`, but there are [many othe
             'I love how amazingly simple ML has become!', 
             'I hate doing mundane and thankless tasks. ☹️'
         ],
-        cache_model => TRUE
+        cache => TRUE
     ) AS positivity;
     ```
 
