How to Make Entity Smaller when its a Baby (Custom Code)

its on forge 1.20.1

i already know to make a entity smaller,so how to make the entity smaller ONLY when its a baby and when it grows it becomes to the original size

also i dont want to make 2 separate entities

(custom code)

(update 09/07: dont need it anymore)