Bram Vanroy commited on
Commit
c4c7f48
1 Parent(s): 50704ab

sort json keys

Browse files
Files changed (1) hide show
  1. generate_overview_json.py +1 -2
generate_overview_json.py CHANGED
@@ -1,6 +1,5 @@
1
  from pathlib import Path
2
  import json
3
- from pprint import pprint
4
 
5
  from transformers import AutoModelForCausalLM
6
 
@@ -45,7 +44,7 @@ def main():
45
  elif "load_in_4bit" in model_args:
46
  results[short_name]["quantization"] = "4-bit"
47
 
48
- pf_overview.write_text(json.dumps(results, indent=4), encoding="utf-8")
49
 
50
 
51
  if __name__ == '__main__':
 
1
  from pathlib import Path
2
  import json
 
3
 
4
  from transformers import AutoModelForCausalLM
5
 
 
44
  elif "load_in_4bit" in model_args:
45
  results[short_name]["quantization"] = "4-bit"
46
 
47
+ pf_overview.write_text(json.dumps(results, indent=4, sort_keys=True), encoding="utf-8")
48
 
49
 
50
  if __name__ == '__main__':