Fix visual bug in quantization (#92)
It didn't calculate the compression ratio properly because of a copy/paste error :(
This commit is contained in:
parent
5b41cd7e5d
commit
09ec3145b3
2
rwkv.cpp
2
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);
|
||||
|
|
Loading…
Reference in New Issue