Using u = Flux.loadmodel!(empty_model, model_state)
leads to the following error

since in Flux v0.14 pad_size is not stored extra (model_state2) whereas in the new version it is (model_state)

Is there a workaround to upgrade old model states?