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

Update preprocessor_inpaint.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 14:57:58 -08:00
parent 0cdc68b7f1
commit 981cee1378
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/forge_preprocessor_inpaint/scripts/preprocessor_inpaint.py
View File

@@ -93,7 +93,7 @@ class PreprocessorInpaintLama(PreprocessorInpaintOnly):
return
def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs):
input_image, remove_pad = resize_image_with_pad(input_image, resolution)
input_image, remove_pad = resize_image_with_pad(input_image, 256)
self.load_model()
@@ -114,7 +114,8 @@ class PreprocessorInpaintLama(PreprocessorInpaintOnly):
result *= 255.0
result = result.detach().cpu().numpy().clip(0, 255).astype(np.uint8)
return remove_pad(result)
result = remove_pad(result)
return result
add_supported_preprocessor(PreprocessorInpaint())