LogitsGenerator -> ModelObj

dottxt-ai/outlines#970
parkervg · Jun 17, 2024 · da799ab · da799ab
1 parent 152fb53
commit da799ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/reference/ingredients/creating-custom-ingredients.md b/docs/reference/ingredients/creating-custom-ingredients.md
@@ -31,11 +31,11 @@ class SummaryProgram(Program):
     """Program to call Model and return summary of the passed table.
     """
 
-    def __call__(self, model: Model, serialized_db: str):  
+    def __call__(self, model: Model, serialized_db: str):
         prompt = f"Summarize the table below.\n\n{serialized_db}\n"
         # Below we follow the outlines pattern for unconstrained text generation
         # https://github.com/outlines-dev/outlines
-        generator = outlines.generate.text(model.logits_generator)
+        generator = outlines.generate.text(model.model_obj)
         # Finally, return (response, prompt) tuple
         # Returning the prompt here allows the underlying BlendSQL classes to track token usage
         return (generator(prompt), prompt)