lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-22 07:43:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update shared.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 12:18:55 -08:00
parent 714d1f0f1b
commit 99ea8e62f5
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules_forge/shared.py
View File

@@ -24,15 +24,16 @@ def add_preprocessor(preprocessor):
class PreprocessorParameter:
def __init__(self, minimum=0.0, maximum=1.0, step=0.01, label='Parameter 1', value=0.5, visible=False):
def __init__(self, minimum=0.0, maximum=1.0, step=0.01, label='Parameter 1', value=0.5, visible=False, **kwargs):
self.gradio_update_kwargs = dict(
minimum=minimum, maximum=maximum, step=step, label=label, value=value, visible=visible
minimum=minimum, maximum=maximum, step=step, label=label, value=value, visible=visible, **kwargs
)
class PreprocessorBase:
def __init__(self):
self.name = 'PreprocessorBase'
self.tag = None
self.slider_1 = PreprocessorParameter()
self.slider_2 = PreprocessorParameter()
self.slider_3 = PreprocessorParameter()
@@ -48,6 +49,9 @@ class PreprocessorBase:
self.model_patcher = ModelPatcher(model=model, load_device=load_device, offload_device=offload_device, **kwargs)
return
def process_before_every_sampling(self, process, cnet):
return
class PreprocessorNone(PreprocessorBase):
def __init__(self):
@@ -62,6 +66,7 @@ class PreprocessorCanny(PreprocessorBase):
def __init__(self):
super().__init__()
self.name = 'canny'
self.tag = 'Canny'
self.slider_1 = PreprocessorParameter(minimum=0, maximum=256, step=1, value=100, label='Low Threshold', visible=True)
self.slider_2 = PreprocessorParameter(minimum=0, maximum=256, step=1, value=200, label='High Threshold', visible=True)