"""Model wrappers for finetuning."""