system_prompt = ( "You are a game theory expert. Solve the following problem step by step. " "Show your reasoning clearly, then provide your final answer.") problem = "Consider a 2-player game with the following payoff matrix: " "L: (3,2) (1,4), R: (2,3) (4,1). Find all Nash Equilibria." messages = [ {"role": "system", "content": system_prompt}, {"role": "user", "content": problem},] prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = tokenizer(prompt, return_tensors="pt").to(model.device) with torch.no_grad(): output = model.generate(**inputs, max_new_tokens=1024, do_sample=False) response = tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)print(response)