The probably cleanest way would be to load the state_dict into the new model definition.
How did you prune the original model?
This might give us some information about the easiest way to load the parameters.