| |
| """Run JSONLite offline with the bundled native MLX implementation. |
| |
| Defaults to model.safetensors beside this script. The verified source bundle |
| includes the exact model architecture, tokenizer, native JSON prompt/decoder and examples. |
| """ |
| from pathlib import Path |
| import hashlib |
| import runpy |
| import sys |
| import tempfile |
| import zipfile |
|
|
| ROOT = Path(__file__).resolve().parent |
| BUNDLE_SHA256 = "8f5a075f63e1ff8857af3ddd916afc57742e614b1da5cbbe15452e9c84e94d2a" |
|
|
| def main(): |
| archive = ROOT / "native_bundle.zip" |
| if not archive.is_file(): |
| raise SystemExit("Missing native_bundle.zip; download the complete repository.") |
| if hashlib.sha256(archive.read_bytes()).hexdigest() != BUNDLE_SHA256: |
| raise SystemExit("Native source bundle failed its integrity check.") |
| if not any(arg == "--weights" or arg.startswith("--weights=") for arg in sys.argv[1:]): |
| sys.argv.extend(["--weights", str(ROOT / "model.safetensors")]) |
| with tempfile.TemporaryDirectory(prefix="jsonlite-runtime-") as directory: |
| runtime = Path(directory) |
| with zipfile.ZipFile(archive) as bundle: |
| for member in bundle.infolist(): |
| path = Path(member.filename) |
| if path.is_absolute() or ".." in path.parts or (member.external_attr >> 16) & 0o170000 == 0o120000: |
| raise SystemExit("Unsafe source-bundle entry.") |
| bundle.extractall(runtime) |
| runpy.run_path(str(runtime / "inference.py"), run_name="__main__") |
|
|
| if __name__ == "__main__": |
| main() |
|
|
