feature: better lite-lm rendering

### Issue

this is very non urgent!

<img width="1028" height="245" alt="Image" src="https://github.com/user-attachments/assets/f3aaab89-77ef-49d6-9e9e-ffa72f54e123" />

### Version and model info

_No response_