from transformers import AutoModelForCausalLM, AutoTokenizer model_name = "perciqa/Perciqa-Aurora-Code" tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForCausalLM.from_pretrained( model_name, torch_dtype="auto", device_map="auto") system_prompt = "You are Aurora, an AI code assistant built by Perciqa. You help developers write, review, and understand code. You provide clear, correct, and complete solutions. When you're unsure, you say so." prompt = "Write a Python function to merge two sorted lists."messages = [ {"role": "system", "content": system_prompt}, {"role": "user", "content": prompt}]text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = tokenizer([text], return_tensors="pt").to(model.device) outputs = model.generate(**inputs, max_new_tokens=2048)print(tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True))