nodes/mcmonkeyprojects-sd-dynamic-thresholding
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

probably version cross-compat on last fix, for #60

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-08-20 03:35:24 -07:00
parent 8b29875902
commit 96238f443e
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/dynamic_thresholding.py
+5 -1
View File
@@ -18,6 +18,10 @@ try:
import dynthres_unipc
except Exception as e:
print(f"\n\n======\nError! UniPC sampler support failed to load! Is your WebUI up to date?\n(Error: {e})\n======")
try:
from modules.sd_samplers_kdiffusion import CFGDenoiserKDiffusion as cfgdenoisekdiff
except Exception as e:
from modules.sd_samplers_kdiffusion import CFGDenoiser as cfgdenoisekdiff
######################### Data values #########################
VALID_MODES = ["Constant", "Linear Down", "Cosine Down", "Half Cosine Down", "Linear Up", "Cosine Up", "Half Cosine Up", "Power Up", "Power Down", "Linear Repeating", "Cosine Repeating", "Sawtooth"]
@@ -189,7 +193,7 @@ class CustomVanillaSDSampler(sd_samplers_compvis.VanillaStableDiffusionSampler):
######################### K-Diffusion Implementation logic #########################
class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiserKDiffusion):
class CustomCFGDenoiser(cfgdenoisekdiff):
def __init__(self, model, dtData):
super().__init__(model)
self.main_class = dtData