import torchfrom transformers import AutoTokenizer, Gemma4UnifiedForConditionalGeneration repo = "qiusizhan/PersistBD-Gemma4-12B-clean-20261022"tokenizer = AutoTokenizer.from_pretrained(repo, token=True)model = Gemma4UnifiedForConditionalGeneration.from_pretrained(repo, token=True, dtype=torch.bfloat16).to("cuda").eval()messages = [{"role": "user", "content": "Reply with OK."}]inputs = tokenizer.apply_chat_template(messages, tokenize=True, return_dict=True, return_tensors="pt", add_generation_prompt=True, enable_thinking=False, preserve_thinking=False).to("cuda")with torch.inference_mode(): output = model.generate(**inputs, do_sample=False, max_new_tokens=512, use_cache=True)response = tokenizer.decode(output[0, inputs["input_ids"].shape[1]:], skip_special_tokens=False)