import json, torchfrom transformers import AutoProcessor, Qwen3_5ForConditionalGenerationfrom huggingface_hub import hf_hub_downloadfrom grid_solver import GridSolver, generate_gridrepo = "omrisap/qwen35-4b-grid-b-50k"processor = AutoProcessor.from_pretrained(repo)contract = json.load(open(hf_hub_download(repo, "solver_contract.json")))base = Qwen3_5ForConditionalGeneration.from_pretrained(repo, dtype=torch.bfloat16).cuda()solver = GridSolver(base, contract["identity"]["allowed_ids"]).eval()# inputs: processor-generated tensors for hybrid alternating demo messages,# query input, enable_thinking=False and add_generation_prompt=True.# generated_ids, terminated = generate_grid(solver, inputs, max_new_tokens=931)