| #!/usr/bin/env python3 | |
| """모델 파일 복원 스크립트""" | |
| import json, base64, glob, os, sys | |
| def reconstruct(hf_base, output_path=None): | |
| parts = sorted(glob.glob(f"{hf_base}.part*.json")) | |
| if not parts: | |
| print(f"파일 없음: {hf_base}.part*.json") | |
| return | |
| if output_path is None: | |
| output_path = hf_base | |
| os.makedirs(os.path.dirname(output_path) or ".", exist_ok=True) | |
| with open(output_path, "wb") as out: | |
| for p in parts: | |
| with open(p) as f: | |
| chunk_data = json.load(f) | |
| out.write(base64.b64decode(chunk_data["data"])) | |
| print(f" {p} → OK") | |
| print(f"복원 완료: {output_path}") | |
| if __name__ == "__main__": | |
| base = sys.argv[1] if len(sys.argv) > 1 else "models/cropdoc_efficientnet_v2/model.pt" | |
| out = sys.argv[2] if len(sys.argv) > 2 else base | |
| reconstruct(base, out) | |