Code review tricky with local models on CPU probably better to think on it lol