from transformers import AutoModelForCausalLM, AutoTokenizer model = AutoModelForCausalLM.from_pretrained( "dracko14/Myth-4B", torch_dtype="auto", device_map="auto", trust_remote_code=True)tokenizer = AutoTokenizer.from_pretrained( "dracko14/Myth-4B", trust_remote_code=True) messages = [ {"role": "system", "content": "You are Myth, a helpful AI assistant."}, {"role": "user", "content": "Write a Python function for binary search."},]prompt = tokenizer.apply_chat_template(messages, tokenize=False)inputs = tokenizer(prompt, return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=200)print(tokenizer.decode(outputs[0], skip_special_tokens=True))