Various docs fixes (dottxt-ai#1429)

Follow up of dottxt-ai#1343 In this PR I added: - all models to the API reference (we had only openai and transformers before) - fixes to the docstrings that were not following the numpy style - fixes to docstrings wrongly rendered by mkdocstring Thanks :)
RobinPicard · Feb 24, 2025 · 02466d3 · 02466d3
1 parent 889364f
commit 02466d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/outlines/models/llamacpp.py b/outlines/models/llamacpp.py
@@ -154,7 +154,7 @@ def generate(self, model_input, logits_processor, **inference_kwargs):
 
         Arguments
         ---------
-        prompt
+        model_input
             A prompt.
         logits_processor
             The logits processor to use when generating text.
@@ -190,7 +190,7 @@ def stream(
 
         Arguments
         ---------
-        prompt
+        model_input
             A prompt.
         logits_processor
             The logits processor to use when generating text.

diff --git a/outlines/models/transformers_vision.py b/outlines/models/transformers_vision.py