File size: 1,040 Bytes
a0a9254 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | """Check the local runtime without loading a second model."""
from pathlib import Path
import hashlib,json,platform,subprocess
import torch,transformers,safetensors
ROOT=Path(__file__).resolve().parents[1]
assert platform.system()=='Darwin' and platform.machine()=='arm64','Native Apple Silicon Python required'
assert torch.backends.mps.is_available(),'PyTorch MPS is unavailable'
expected='0a93cfcf7121112fdf90336bda1198ad67a3bb45247da6eedf0dd8672d4d53b5'
h=hashlib.sha256()
with (ROOT/'model/best.safetensors').open('rb') as f:
for block in iter(lambda:f.read(8*1024*1024),b''):h.update(block)
assert h.hexdigest()==expected,'Model checksum mismatch'
browser=subprocess.check_output([str(ROOT/'node_modules/.bin/agent-browser'),'--version'],text=True).strip()
assert browser=='agent-browser 0.37.1',browser
print(json.dumps({'passed':True,'platform':platform.platform(),'torch':torch.__version__,'transformers':transformers.__version__,'safetensors':safetensors.__version__,'agent_browser':browser,'modelSha256':expected},indent=2))
|