from transformers import AutoModelForCausalLM, AutoTokenizer model_id = "zbeeb/deepseek-r1-distill-qwen-14b-fast-math-r1-sft-10ep" tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype="auto", device_map="auto", trust_remote_code=True,) messages = [ {"role": "system", "content": "Please reason step by step, and put your final answer within \\boxed{}."}, {"role": "user", "content": "What is 17 * 23?"},] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, return_tensors="pt",).to(model.device) outputs = model.generate( inputs, max_new_tokens=2048, temperature=0.6, top_p=0.95, do_sample=True,) print(tokenizer.decode(outputs[0], skip_special_tokens=True))