Custom constrained decoding could have solved this. Penalize comment tokens :)