From 22fa5a621fa2f8249943e0a52dd8c8a21e9baca0 Mon Sep 17 00:00:00 2001
From: Andrei Betlen <abetlen@gmail.com>
Date: Thu, 13 Apr 2023 00:19:55 -0400
Subject: [PATCH] Revert "Deprecate generate method"

This reverts commit 6cf58765388b85329769cb78405c5e5ff74dc414.
---
 llama_cpp/llama.py | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
index c545420..67fefe5 100644
--- a/llama_cpp/llama.py
+++ b/llama_cpp/llama.py
@@ -3,7 +3,6 @@ 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
@@ -240,11 +239,6 @@ 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: