| # Script to rename checkpoint from dp to deep_phonemizer | |
| import io, zipfile, torch | |
| old = "model_step_420k.pt" | |
| new = "model_step_420k_fixed.pt" | |
| # Step 1. Read zip members | |
| with open(old, "rb") as f: | |
| data = f.read() | |
| with zipfile.ZipFile(io.BytesIO(data)) as zin: | |
| buf = io.BytesIO() | |
| with zipfile.ZipFile(buf, "w", compression=zipfile.ZIP_DEFLATED) as zout: | |
| for item in zin.infolist(): | |
| file_bytes = zin.read(item.filename) | |
| file_bytes = file_bytes.replace(b"dp.", b"deep_phonemizer.") | |
| zout.writestr(item, file_bytes) | |
| # Step 2. Save the modified zip | |
| with open(new, "wb") as f: | |
| f.write(buf.getvalue()) | |
| # Step 3. Verify | |
| checkpoint = torch.load(new, map_location="cpu", weights_only=False) | |
| print("✅ Loaded successfully") | |