From bc17c1ac7fcae41e1b81d218e064b18835ab9961 Mon Sep 17 00:00:00 2001 From: "M. Eren Akbiyik" Date: Sat, 2 Sep 2023 00:44:57 +0200 Subject: [PATCH] fix: bug in compression --- torchcache/torchcache.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchcache/torchcache.py b/torchcache/torchcache.py index 81a86a4..3c38f1a 100644 --- a/torchcache/torchcache.py +++ b/torchcache/torchcache.py @@ -548,8 +548,8 @@ def _cache_to_file(self, embedding: Tensor, hash_val: int) -> None: if self.zstd_compression: raw_data = zstd.compress( raw_data, - level=self.zstd_compression_level, - threads=self.zstd_compression_threads, + self.zstd_compression_level, + self.zstd_compression_threads, ) logger.debug(f"Compressed data size: {len(raw_data)}")