Skip to content

Commit b17af98

Browse files
committed
Fix typo
1 parent 89164cb commit b17af98

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/deepsparse/v2/text_generation/prep_for_single_engine.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def can_operate(self, inp: Any, context: Context, inference_state: InferenceStat
3939
kv_cache = inp.get("kv_cache")
4040
tokens = inp.get("tokens")
4141
# if 0 prompt tokens remain, can't operate (multi-token engine has already run)
42-
if len(tokens) == kv_cache.total_num_processed_tokens == 0:
42+
if len(tokens) == kv_cache.total_num_processed_tokens:
4343
return False
4444

4545
# if number of prompt tokens left to process is >= self.prompt_sequnce_length

0 commit comments

Comments
 (0)