lvkaokao commited on
Commit
6b27724
1 Parent(s): c2b975e

Update src/leaderboard/read_evals.py

Browse files
Files changed (1) hide show
  1. src/leaderboard/read_evals.py +2 -2
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)