import torchfrom peft import PeftModelfrom transformers import AutoProcessor, Gemma4ForConditionalGeneration base = "google/gemma-4-E4B-it"processor = AutoProcessor.from_pretrained(base)model = Gemma4ForConditionalGeneration.from_pretrained(base, torch_dtype=torch.bfloat16, device_map="auto")model = PeftModel.from_pretrained(model, "NearAndDear98/Audio-R1-Gemma-4-E4B-it").eval() SYSTEM = ( "Your response MUST contain two parts in this exact order:\n" "1. Your reasoning enclosed within <think> </think> tags.\n" "2. The final answer — a single letter for the correct option — enclosed " "within <answer> </answer> tags.")messages = [ {"role": "system", "content": SYSTEM}, {"role": "user", "content": [ {"type": "text", "text": "Question ...\n(A) ...\n(B) ...\n(C) ...\n(D) ..."}, {"type": "audio", "audio": "example.wav"}, # audio after text, per Gemma 4 docs ]},]inputs = processor.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", enable_thinking=False,).to(model.device)out = model.generate(**inputs, max_new_tokens=1024, do_sample=False)print(processor.decode(out[0, inputs["input_ids"].shape[-1]:], skip_special_tokens=True))