Write the _process_model_after_weight_loading method.