switch cfg to args["denoised"]

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

View File

@ -21,7 +21,7 @@ class Mahiro:
leap = cond_p * scale
#sim with uncond leap
u_leap = uncond_p * scale
cfg = uncond_p + (cond_p - uncond_p) * scale
cfg = args["denoised"]
merge = (leap + cfg) / 2
normu = torch.sqrt(u_leap.abs()) * u_leap.sign()
normm = torch.sqrt(merge.abs()) * merge.sign()