Fix example documentation

abetlen · abetlen · commit eef627c09c16 · 2023-04-01T17:39:35.000-04:00
diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
@@ -127,10 +127,11 @@ def generate(
     ]:
         """Generate tokens.
 
-        >>> llama = Llama("models/117M")
-        >>> tokens = llama.tokenize(b"Hello, world!")
-        >>> for token in llama.generate(tokens, top_k=40, top_p=0.95, temp=1.0, repeat_penalty=1.1):
-        ...     print(llama.detokenize([token]))
+        Examples:
+            >>> llama = Llama("models/ggml-7b.bin")
+            >>> tokens = llama.tokenize(b"Hello, world!")
+            >>> for token in llama.generate(tokens, top_k=40, top_p=0.95, temp=1.0, repeat_penalty=1.1):
+            ...     print(llama.detokenize([token]))
 
         Args:
             tokens: The prompt tokens.
