Advanced Model Sampling

[Mol31]

Thank you for your hard work on mainline Forge and this extension.

With the new developments in local AI models, there is a shift towards Vpred, ZSNR-trained models. Is it possible for this extension to adapt Advanced Model Sampling from the Reforge fork?

Many thanks for considering my request.

[DenOfEquity]

Looking at the reforge AMS builtin extension, it is based on ldm_patched (comfy) code which was completely stripped out in lllyasviel's forge2 update. The missing parts seem to be:

• zero-terminal snr sigmas, which forge2 has a setting for. Though I'm not sure how effective it is, so this should be looked at. Simply copying over the function and applying it to the sigmas does not work.
• rescaleCFG, which is included in the LatentModifer Integrated extension, as Rescale Cfg Phi (the slider has much too high a range, typical best value is around 0.7)
• (maybe) allowing relatively very high sigma_max, which can be set in the Settings; the allowed range probably needs to be increased. Though not all schedulers use that setting, the older ones do.

[Enferlain]

k_prediction.py seems to have has stuff pertaining to v-pred, edm, epsilon etc (and now ztsnr) but for some reason I can't get cosxl to work even if I try selecting it in your extension, weird

I did confirm v-pred works though so that's nice