revert the sampler_cfg_function thing

This commit is contained in:
yoinked 2024-12-10 08:23:11 -08:00 committed by GitHub
parent 61a875f0b2
commit 144255a06d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -28,8 +28,8 @@ class Mahiro:
sim = F.cosine_similarity(normu, normm).mean()
simsc = 2 * (sim+1)
wm = (simsc*cfg + (4-simsc)*leap) / 4
return args['input'] - wm
m.set_model_sampler_cfg_function(mahiro_normd)
return wm
m.set_model_sampler_post_cfg_function(mahiro_normd)
return (m, )
NODE_CLASS_MAPPINGS = {