Update src/leaderboard/read_evals.py
Browse files
src/leaderboard/read_evals.py
CHANGED
@@ -60,8 +60,8 @@ class EvalResult:
|
|
60 |
weight_dtype = WeightDtype.from_str(data["task_info"].get("weight_dtype", "int4"))
|
61 |
compute_dtype = ComputeDtype.from_str(data["task_info"].get("compute_dtype", "bfloat16"))
|
62 |
# double_quant = data["quantization_config"].get("bnb_4bit_use_double_quant", False)
|
63 |
-
model_params = round(config["model_params"], 2)
|
64 |
-
model_size = round(config["model_size"], 2)
|
65 |
# group_size = data["quantization_config"].get("group_size", -1)
|
66 |
if data.get("quantization_config", None):
|
67 |
double_quant = data["quantization_config"].get("bnb_4bit_use_double_quant", False)
|
|
|
60 |
weight_dtype = WeightDtype.from_str(data["task_info"].get("weight_dtype", "int4"))
|
61 |
compute_dtype = ComputeDtype.from_str(data["task_info"].get("compute_dtype", "bfloat16"))
|
62 |
# double_quant = data["quantization_config"].get("bnb_4bit_use_double_quant", False)
|
63 |
+
model_params = round(float(config["model_params"]), 2)
|
64 |
+
model_size = round(float(config["model_size"]), 2)
|
65 |
# group_size = data["quantization_config"].get("group_size", -1)
|
66 |
if data.get("quantization_config", None):
|
67 |
double_quant = data["quantization_config"].get("bnb_4bit_use_double_quant", False)
|