As an ent, and in my opinion, any punishment for catching him smoking weed besides talking to him about responsible consumption is disciplining him too harshly.

EDIT: I stick to my OPINION, let the downvotes keep coming. It is an opinion, as stated.