import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_id = "petergilani/Macaron-V1-Tall-L2-Coding-Fused-BF16" tokenizer = AutoTokenizer.from_pretrained( model_id, trust_remote_code=True,) model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True,)model.eval() messages = [ {"role": "user", "content": "Write a Python function that performs a binary search on a sorted list and returns the index of the target."}] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, return_tensors="pt",).to(model.device) generated_ids = model.generate( **inputs, max_new_tokens=512, do_sample=True, temperature=0.2,)print(tokenizer.decode(generated_ids[0][inputs["input_ids"].shape[-1]:], skip_special_tokens=True))