diff options
| author | timntorres <116157310+timntorres@users.noreply.github.com> | 2022-11-02 09:18:21 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-02 09:18:21 +0000 |
| commit | 9c67408004ed132637d10321bf44565f82055fd2 (patch) | |
| tree | 2cf1cf92928cbb3de02f511d35e5ba1f1146a927 /modules/sd_samplers.py | |
| parent | 4a8cf01f6f7f072cc9c67d6b31662384b212dd9c (diff) | |
| download | stable-diffusion-webui-gfx803-9c67408004ed132637d10321bf44565f82055fd2.tar.gz stable-diffusion-webui-gfx803-9c67408004ed132637d10321bf44565f82055fd2.tar.bz2 stable-diffusion-webui-gfx803-9c67408004ed132637d10321bf44565f82055fd2.zip | |
Allow saving "before-highres-fix. (#4150)
* Save image/s before doing highres fix.
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 44d4c189..d7fa89a0 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -93,9 +93,8 @@ def single_sample_to_image(sample): return Image.fromarray(x_sample)
-def sample_to_image(samples):
- return single_sample_to_image(samples[0])
-
+def sample_to_image(samples, index=0):
+ return single_sample_to_image(samples[index])
def samples_to_image_grid(samples):
return images.image_grid([single_sample_to_image(sample) for sample in samples])
|