import torchfrom transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfigfrom peft import PeftModel BASE = "Qwen/Qwen3-14B"ADAPTER = "cs-552-2026-Clanker-Scientists/coordinator-qwen3-14b-qlora-legal-v3" tok = AutoTokenizer.from_pretrained(BASE)bnb = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_compute_dtype=torch.bfloat16)base = AutoModelForCausalLM.from_pretrained(BASE, quantization_config=bnb, device_map="auto")model = PeftModel.from_pretrained(base, ADAPTER)model.eval() msgs = [ {"role": "system", "content": "You are a legal contract analyst..."}, {"role": "user", "content": "Contract:\n[excerpt]\n\nQuestion: Does the contract explicitly identify confidential information?"},]prompt = tok.apply_chat_template(msgs, tokenize=False, add_generation_prompt=True, enable_thinking=True)inputs = tok(prompt, return_tensors="pt").to(model.device)with torch.no_grad(): out = model.generate(**inputs, max_new_tokens=512, do_sample=False)print(tok.decode(out[0, inputs.input_ids.shape[1]:], skip_special_tokens=True))