Fix visual bug in quantization (#92)

LoganDark · web-flow · commit 09ec3145b315 · 2023-06-07T16:45:21.000+05:00
It didn't calculate the compression ratio properly because of a
copy/paste error :(
diff --git 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);

Original file line number	Diff line number	Diff line change
`@@ -1264,7 +1264,7 @@ bool rwkv_quantize_model_file(const char * in_path, const char * out_path, const`
`1264`	`1264`
`1265`	`1265`	`RWKV_ASSERT_FALSE_MSG(RWKV_ERROR_FILE_WRITE, rwkv_fwrite_tensor(out_file, tensor), "Failed to write tensor %s", name_str);`
`1266`	`1266`	`orig_total_size += orig_size;`
`1267`		`- new_total_size += orig_size;`
	`1267`	`+ new_total_size += new_size;`
`1268`	`1268`	`}`
`1269`	`1269`
`1270`	`1270`	`RWKV_MSG("original size = %8.2f MB\n", orig_total_size / 1024.0 / 1024.0);`