mp: Fix checkpoint saving (Comfy-Org#12268)

Fix regression in the recent model saving refactor. Pass the non unet
pieces down the layers so that checkpoints are complete.

Author: rattus128

comfy/model_patcher.py

@@ -1400,7 +1400,7 @@ def state_dict_for_saving(self, clip_state_dict=None, vae_state_dict=None, clip_
                 continue
             key = "diffusion_model." + k
             unet_state_dict[k] = LazyCastingParam(self, key, comfy.utils.get_attr(self.model, key))
-        return self.model.state_dict_for_saving(unet_state_dict)
+        return self.model.state_dict_for_saving(unet_state_dict, clip_state_dict=clip_state_dict, vae_state_dict=vae_state_dict, clip_vision_state_dict=clip_vision_state_dict)
 
     def __del__(self):
         self.unpin_all_weights()