Updates model to init function space before init_variables