diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py index 3e13776..69f7680 100644 --- a/llama_cpp/llama.py +++ b/llama_cpp/llama.py @@ -3,6 +3,7 @@ import sys import uuid import time import math +import warnings import multiprocessing from typing import List, Optional, Union, Generator, Sequence, Iterator from collections import deque @@ -239,6 +240,11 @@ class Llama: Yields: The generated tokens. """ + warnings.warn( + "Llama.generate is deprecated and will be removed in v0.2.0", + DeprecationWarning, + stacklevel=2, + ) assert self.ctx is not None self.reset() while True: