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

update for new upstream WebUI sampler reorg, for #2

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2023-01-30 09:27:35 -08:00
parent 4a36b4e7ac
commit b5a7983454
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/dynamic_thresholding.py
+4 -4
View File
@@ -4,7 +4,7 @@
# Author: Alex 'mcmonkey' Goodwin # Author: Alex 'mcmonkey' Goodwin
# GitHub URL: https://github.com/mcmonkeyprojects/sd-dynamic-thresholding # GitHub URL: https://github.com/mcmonkeyprojects/sd-dynamic-thresholding
# Created: 2022/01/26 # Created: 2022/01/26
# Last updated: 2023/01/26 # Last updated: 2023/01/30
# #
# For usage help, view the README.md file in the extension root, or via the GitHub page. # For usage help, view the README.md file in the extension root, or via the GitHub page.
# #
@@ -14,7 +14,7 @@ import gradio as gr
import random import random
import torch import torch
import math import math
from modules import sd_samplers, scripts from modules import scripts, sd_samplers, sd_samplers_kdiffusion, sd_samplers_common
######################### Data values ######################### ######################### Data values #########################
VALID_MODES = ["Constant", "Linear Down", "Cosine Down", "Linear Up", "Cosine Up"] VALID_MODES = ["Constant", "Linear Down", "Cosine Down", "Linear Up", "Cosine Up"]
@@ -71,7 +71,7 @@ class Script(scripts.Script):
cfg = CustomCFGDenoiser(result.model_wrap_cfg.inner_model, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, p.steps) cfg = CustomCFGDenoiser(result.model_wrap_cfg.inner_model, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, p.steps)
result.model_wrap_cfg = cfg result.model_wrap_cfg = cfg
return result return result
newSampler = sd_samplers.SamplerData(fixed_sampler_name, newConstructor, sampler.aliases, sampler.options) newSampler = sd_samplers_common.SamplerData(fixed_sampler_name, newConstructor, sampler.aliases, sampler.options)
# Apply for usage # Apply for usage
p.orig_sampler_name = p.sampler_name p.orig_sampler_name = p.sampler_name
p.sampler_name = fixed_sampler_name p.sampler_name = fixed_sampler_name
@@ -88,7 +88,7 @@ class Script(scripts.Script):
######################### Implementation logic ######################### ######################### Implementation logic #########################
class CustomCFGDenoiser(sd_samplers.CFGDenoiser): class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiser):
def __init__(self, model, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, maxSteps): def __init__(self, model, mimic_scale, threshold_percentile, mimic_mode, cfg_mode, maxSteps):
super().__init__(model) super().__init__(model)
self.mimic_scale = mimic_scale self.mimic_scale = mimic_scale