Bram Vanroy
commited on
Commit
•
c4c7f48
1
Parent(s):
50704ab
sort json keys
Browse files
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__':
|