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

Fix Auto1.6 compat

Browse Source 
I love auto updates that change and break things without warning. For #69
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-09-09 23:57:33 -07:00
parent 96238f443e
commit 55ca687fb8
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/dynamic_thresholding.py
+5 -2
View File
@@ -13,15 +13,18 @@
import gradio as gr import gradio as gr
import torch, traceback import torch, traceback
import dynthres_core import dynthres_core
from modules import scripts, script_callbacks, sd_samplers, sd_samplers_compvis, sd_samplers_kdiffusion, sd_samplers_common from modules import scripts, script_callbacks, sd_samplers, sd_samplers_compvis, sd_samplers_common
try: try:
import dynthres_unipc import dynthres_unipc
except Exception as e: 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======") print(f"\n\n======\nError! UniPC sampler support failed to load! Is your WebUI up to date?\n(Error: {e})\n======")
try: try:
from modules.sd_samplers_kdiffusion import CFGDenoiserKDiffusion as cfgdenoisekdiff from modules.sd_samplers_kdiffusion import CFGDenoiserKDiffusion as cfgdenoisekdiff
IS_AUTO_16 = True
except Exception as e: except Exception as e:
print(f"\n\n======\nWarning! Using legacy KDiff version! Is your WebUI up to date?\n======")
from modules.sd_samplers_kdiffusion import CFGDenoiser as cfgdenoisekdiff from modules.sd_samplers_kdiffusion import CFGDenoiser as cfgdenoisekdiff
IS_AUTO_16 = False
######################### Data values ######################### ######################### 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"] 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"]
@@ -145,7 +148,7 @@ class Script(scripts.Script):
else: else:
def newConstructor(model): def newConstructor(model):
result = sampler.constructor(model) result = sampler.constructor(model)
cfg = CustomCFGDenoiser(result.model_wrap_cfg.inner_model, dtData) cfg = CustomCFGDenoiser(result if IS_AUTO_16 else result.model_wrap_cfg.inner_model, dtData)
result.model_wrap_cfg = cfg result.model_wrap_cfg = cfg
return result return result
newSampler = sd_samplers_common.SamplerData(fixed_sampler_name, newConstructor, sampler.aliases, sampler.options) newSampler = sd_samplers_common.SamplerData(fixed_sampler_name, newConstructor, sampler.aliases, sampler.options)