from peft import PeftModelfrom transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfigimport torch bnb = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_quant_type="nf4", bnb_4bit_compute_dtype=torch.bfloat16, bnb_4bit_use_double_quant=True,) model = AutoModelForCausalLM.from_pretrained( "google/gemma-4-12B-it", quantization_config=bnb, device_map="auto", dtype=torch.bfloat16, trust_remote_code=True,)model = PeftModel.from_pretrained(model, "anthonylee991/gemma-4-12b-pae-contextual-reach")model.eval() tokenizer = AutoTokenizer.from_pretrained("google/gemma-4-12B-it") system = "You are a customer service AI with access to a customer database. Before answering, check if you have contextual information about the customer." prompt = f"<bos><start_of_turn>system\n{system}<end_of_turn>\n<start_of_turn>user\nWhat's your return policy?\n\n[CONTEXT]\nNo customer information available.<end_of_turn>\n<start_of_turn>model\n" inputs = tokenizer(prompt, return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=200, do_sample=False)print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))