dyslexic-writer-qwen3-4b

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("jburnford/dyslexic-writer-qwen3-4b")
tokenizer = AutoTokenizer.from_pretrained("jburnford/dyslexic-writer-qwen3-4b")

messages = [
    {"role": "system", "content": "You are a spelling correction assistant."},
    {"role": "user", "content": "Fix any spelling mistakes in this text. If there are no mistakes, output the text unchanged.\n\nI went to teh store."},
]

prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True, enable_thinking=False)
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256, do_sample=False)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))