File size: 1,559 Bytes
c798f8d 96ec8c0 c798f8d 319dca0 c798f8d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | #!/usr/bin/env python3
"""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()
|