remove python 3.10 code, for #1
parent
3fe3e96543
commit
4a36b4e7ac
|
|
@ -0,0 +1 @@
|
|||
__pycache__/
|
||||
|
|
@ -103,32 +103,26 @@ class CustomCFGDenoiser(sd_samplers.CFGDenoiser):
|
|||
|
||||
def dynthresh(self, cond, uncond, cfgScale, conds_list):
|
||||
mimicScale = self.mimic_scale
|
||||
match self.mimic_mode:
|
||||
case "Constant":
|
||||
if self.mimic_mode == "Constant":
|
||||
pass
|
||||
case "Linear Down":
|
||||
elif self.mimic_mode == "Linear Down":
|
||||
mimicScale *= 1.0 - (self.step / self.maxSteps)
|
||||
case "Cosine Down":
|
||||
elif self.mimic_mode == "Cosine Down":
|
||||
mimicScale *= 1.0 - math.cos(self.step / self.maxSteps)
|
||||
case "Linear Up":
|
||||
elif self.mimic_mode == "Linear Up":
|
||||
mimicScale *= self.step / self.maxSteps
|
||||
pass
|
||||
case "Cosine Up":
|
||||
elif self.mimic_mode == "Cosine Up":
|
||||
mimicScale *= math.cos(self.step / self.maxSteps)
|
||||
if self.cfg_mode == "Constant":
|
||||
pass
|
||||
match self.cfg_mode:
|
||||
case "Constant":
|
||||
pass
|
||||
case "Linear Down":
|
||||
elif self.cfg_mode == "Linear Down":
|
||||
cfgScale *= 1.0 - (self.step / self.maxSteps)
|
||||
case "Cosine Down":
|
||||
elif self.cfg_mode == "Cosine Down":
|
||||
cfgScale *= 1.0 - math.cos(self.step / self.maxSteps)
|
||||
case "Linear Up":
|
||||
elif self.cfg_mode == "Linear Up":
|
||||
cfgScale *= self.step / self.maxSteps
|
||||
pass
|
||||
case "Cosine Up":
|
||||
elif self.cfg_mode == "Cosine Up":
|
||||
cfgScale *= math.cos(self.step / self.maxSteps)
|
||||
pass
|
||||
# uncond shape is (batch, 4, height, width)
|
||||
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"
|
||||
|
|
|
|||
Loading…
Reference in New Issue