Update preprocessor_inpaint.py

This commit is contained in:
lllyasviel
2024-01-30 13:26:08 -08:00
parent 0ed73f32f2
commit 9b9f301ba2

View File

@@ -16,9 +16,9 @@ class PreprocessorInpaintOnly(PreprocessorInpaint):
def __init__(self):
super().__init__()
self.name = 'inpaint_only'
self.tags = ['Inpaint']
self.model_filename_filters = ['inpaint']
self.slider_resolution = PreprocessorParameter(visible=False)
def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs):
return input_image
def process_before_every_sampling(self, process, cond, *args, **kwargs):
return
@@ -31,9 +31,9 @@ class PreprocessorInpaintLama(PreprocessorInpaintOnly):
def __init__(self):
super().__init__()
self.name = 'inpaint_only+lama'
self.tags = ['Inpaint']
self.model_filename_filters = ['inpaint']
self.slider_resolution = PreprocessorParameter(visible=False)
def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs):
return input_image
add_supported_preprocessor(PreprocessorInpaint())