diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-09-26 14:29:50 +0000 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-09-26 14:29:50 +0000 |
| commit | 740070ea9cdb254209f66417418f2a4af8b099d6 (patch) | |
| tree | 52896a6159b706024af9520c855c10091162372c /modules/extras.py | |
| parent | bfb7f15d46048f27338eeac3a591a5943d03c5f1 (diff) | |
| download | stable-diffusion-webui-gfx803-740070ea9cdb254209f66417418f2a4af8b099d6.tar.gz stable-diffusion-webui-gfx803-740070ea9cdb254209f66417418f2a4af8b099d6.tar.bz2 stable-diffusion-webui-gfx803-740070ea9cdb254209f66417418f2a4af8b099d6.zip | |
Re-implement universal model loading
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index 382ffa7d..4c95cf76 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -36,6 +36,8 @@ def run_extras(extras_mode, image, image_folder, gfpgan_visibility, codeformer_v outputs = []
for image, image_name in zip(imageArr, imageNameArr):
+ if image is None:
+ return outputs, "Please select an input image.", ''
existing_pnginfo = image.info or {}
image = image.convert("RGB")
|