lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update shared.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 10:27:43 -08:00
parent 2c64478e10
commit c5cf030062
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
modules_forge/shared.py
View File

@@ -13,6 +13,13 @@ os.makedirs(preprocessor_dir, exist_ok=True)
shared_preprocessors = {} shared_preprocessors = {}
def add_preprocessor(preprocessor):
global shared_preprocessors
p = preprocessor()
shared_preprocessors[p.name] = p
return
class PreprocessorParameter: 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):
self.gradio_update_kwargs = dict( self.gradio_update_kwargs = dict(
@@ -40,9 +47,6 @@ class PreprocessorNone(PreprocessorBase):
return input_image return input_image
shared_preprocessors['none'] = PreprocessorNone()
class PreprocessorCanny(PreprocessorBase): class PreprocessorCanny(PreprocessorBase):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
@@ -55,4 +59,5 @@ class PreprocessorCanny(PreprocessorBase):
return canny_image return canny_image
shared_preprocessors['canny'] = PreprocessorCanny() add_preprocessor(PreprocessorNone)
add_preprocessor(PreprocessorCanny)