Spaces:
Running
Running
| """Écrit config.json (preset par défaut ou celui demandé) à la racine du dépôt.""" | |
| import argparse | |
| import json | |
| import sys | |
| from pathlib import Path | |
| sys.path.insert(0, str(Path(__file__).resolve().parents[1])) | |
| from mouh.model import DEFAULT_PRESET, PRESETS, estimate_params, preset_config # noqa: E402 | |
| def main() -> None: | |
| ap = argparse.ArgumentParser() | |
| ap.add_argument("--preset", default=DEFAULT_PRESET, choices=sorted(PRESETS)) | |
| ap.add_argument("--out", default=str(Path(__file__).resolve().parents[1] / "config.json")) | |
| a = ap.parse_args() | |
| cfg = preset_config(a.preset) | |
| d = cfg.to_dict() | |
| d.pop("transformers_version", None) | |
| d["mouh_preset"] = a.preset | |
| d["mouh_parameter_report"] = estimate_params(cfg).as_dict() | |
| Path(a.out).write_text(json.dumps(d, indent=2, sort_keys=True) + "\n", encoding="utf-8") | |
| print(f"écrit {a.out}") | |
| if __name__ == "__main__": | |
| main() | |