from mlx_lm import load, generate

model, tokenizer = load("mlx-community/T-pro-it-1.0-Q4-mlx")

prompt="Напиши стих про машинное обучение"

if hasattr(tokenizer, "apply_chat_template") and tokenizer.chat_template is not None:
    messages = [
        {"role": "system", "content": "Ты T-pro, виртуальный ассистент в Т-Технологии. Твоя задача - быть полезным диалоговым ассистентом."},
        {"role": "user", "content": prompt}
    ]
    prompt = tokenizer.apply_chat_template(
        messages, tokenize=False, add_generation_prompt=True
    )

response = generate(model, tokenizer, prompt=prompt, verbose=True)