Revert "Deprecate generate method"

This reverts commit 6cf5876538.
This commit is contained in:
Andrei Betlen 2023-04-13 00:19:55 -04:00
parent 4f5f99ef2a
commit 22fa5a621f

View file

@ -3,7 +3,6 @@ import sys
import uuid import uuid
import time import time
import math import math
import warnings
import multiprocessing import multiprocessing
from typing import List, Optional, Union, Generator, Sequence, Iterator from typing import List, Optional, Union, Generator, Sequence, Iterator
from collections import deque from collections import deque
@ -240,11 +239,6 @@ class Llama:
Yields: Yields:
The generated tokens. 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 assert self.ctx is not None
self.reset() self.reset()
while True: while True: