1580158
/
sd-dynamic-thresholding
mirror of https://github.com/mcmonkeyprojects/sd-dynamic-thresholding.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

additional minor cleaning

pull/87/head
Alex "mcmonkey" Goodwin 2023-10-29 07:26:14 -07:00
parent 450983c621
commit 9fe47f4e7a
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dynthres_core.py
View File

@ -55,9 +55,9 @@ class DynThresh:
scale += min scale += min
return scale return scale
def dynthresh(self, cond, uncond, cfgScale, weights): def dynthresh(self, cond, uncond, cfg_scale, weights):
mimicScale = self.interpretScale(self.mimic_scale, self.mimic_mode, self.mimic_scale_min) mimic_scale = self.interpretScale(self.mimic_scale, self.mimic_mode, self.mimic_scale_min)
cfgScale = self.interpretScale(cfgScale, self.cfg_mode, self.cfg_scale_min) cfg_scale = self.interpretScale(cfg_scale, self.cfg_mode, self.cfg_scale_min)
# uncond shape is (batch, 4, height, width) # uncond shape is (batch, 4, height, width)
conds_per_batch = cond.shape[0] / uncond.shape[0] conds_per_batch = cond.shape[0] / uncond.shape[0]
assert conds_per_batch == int(conds_per_batch), "Expected # of conds per batch to be constant across batches" assert conds_per_batch == int(conds_per_batch), "Expected # of conds per batch to be constant across batches"
@ -70,8 +70,8 @@ class DynThresh:
relative = diff.sum(1) relative = diff.sum(1)
### Get the normal result for both mimic and normal scale ### Get the normal result for both mimic and normal scale
mim_target = uncond + relative * mimicScale mim_target = uncond + relative * mimic_scale
cfg_target = uncond + relative * cfgScale cfg_target = uncond + relative * cfg_scale
### If we weren't doing mimic scale, we'd just return cfg_target here ### If we weren't doing mimic scale, we'd just return cfg_target here
### Now recenter the values relative to their average rather than absolute, to allow scaling from average ### Now recenter the values relative to their average rather than absolute, to allow scaling from average