Skip to content

Commit df90375

Browse files
committed
[fix]clean code fix
1 parent 773388d commit df90375

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

ucm/sandbox/sparse/retake/retake/longvideo_cache.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -488,7 +488,7 @@ def update(
488488
no_compression = math.isclose(compression_ratio, 1.0, rel_tol=1e-9)
489489
# print('compression_ratio of layer %d: %.4f' % (layer_idx, compression_ratio))
490490

491-
if (self.kvcache_compression and self.compression_ratio < 1.0 and no_compression):
491+
if self.kvcache_compression and self.compression_ratio < 1.0 and no_compression:
492492
# Truncate the prompts directly when no compression
493493
key_states = key_states[:, :, : -self.prompt_length]
494494
value_states = value_states[:, :, : -self.prompt_length]

0 commit comments

Comments
 (0)