Skip to content

Commit 6b0cabd

Browse files
committed
Fix green generation labels
1 parent 8260d85 commit 6b0cabd

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

guidance/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "0.1.8"
1+
__version__ = "0.1.9"
22

33
import functools
44
import os

guidance/models/_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -654,7 +654,6 @@ def __call__(self, grammar, max_tokens=1000000, n=1, top_p=1, temperature=0.0, e
654654
token_count = 0
655655
last_token_count = 0
656656
was_forced = False
657-
is_generated = False
658657
captured_data = {}
659658
captured_log_prob_data = {}
660659
while True: # each iteration generates one more token (and some of the associated bytes)
@@ -670,6 +669,7 @@ def __call__(self, grammar, max_tokens=1000000, n=1, top_p=1, temperature=0.0, e
670669
parser.mark_new_token()
671670

672671
# walk down the trie as far as possible before computing the logits
672+
is_generated = False
673673
retry_token_gen = False
674674
trie = self._token_trie
675675
trie.match_version += 1 # this invalidates all the match caches from the previous token

0 commit comments

Comments
 (0)