"""New public export reader, not a historical training-run source snapshot."""
import json,pathlib,sys,torch
from transformers import GPT2Config,GPT2LMHeadModel
from safetensors.torch import load_file
def load(directory):
    directory=pathlib.Path(directory)
    net=GPT2LMHeadModel(GPT2Config.from_dict(json.loads((directory/'config.json').read_text())))
    net.load_state_dict(load_file(str(directory/'model.safetensors')),strict=True)
    return net.eval()
if __name__=='__main__':
    torch.set_num_threads(2)
    net=load(sys.argv[1])
    with torch.inference_mode():
        logits=net(torch.tensor([[69]]),use_cache=False).logits[0,-1].float()
    assert logits.shape==(128,) and torch.isfinite(logits).all()
    print('Loaded FP32 model; 128 finite raw next-token logits from BOS. No move filtering or search applied.')