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__':
|