minor updates to pretrain state and adding set input on finetune phase