GitLab

* Add type annotation UnetWrapperFunction

* nit

* Add types.py

The previous fix didn't cover the case where the model was loaded in
lowvram mode right before.

The old set_attr has been renamed to set_attr_param.

Moved all the sag related code to comfy_extras/nodes_sag.py

Unload models before getting weights for model patching.

This allows changing the sampling parameters of the model (eps or vpred)
or set the model to use zsnr.

Found while tinkering.