From 09ec3145b315a6aec055867a34383c58de8ffc48 Mon Sep 17 00:00:00 2001 From: LoganDark Date: Wed, 7 Jun 2023 04:45:21 -0700 Subject: [PATCH] Fix visual bug in quantization (#92) It didn't calculate the compression ratio properly because of a copy/paste error :( --- rwkv.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rwkv.cpp b/rwkv.cpp index f91fa4f..0d4a548 100644 --- a/rwkv.cpp +++ b/rwkv.cpp @@ -1264,7 +1264,7 @@ bool rwkv_quantize_model_file(const char * in_path, const char * out_path, const RWKV_ASSERT_FALSE_MSG(RWKV_ERROR_FILE_WRITE, rwkv_fwrite_tensor(out_file, tensor), "Failed to write tensor %s", name_str); orig_total_size += orig_size; - new_total_size += orig_size; + new_total_size += new_size; } RWKV_MSG("original size = %8.2f MB\n", orig_total_size / 1024.0 / 1024.0);