Allow saving "before-highres-fix. (#4150)

* Save image/s before doing highres fix.
This commit is contained in:
timntorres
2022-11-02 02:18:21 -07:00
committed by GitHub
parent 4a8cf01f6f
commit 9c67408004
3 changed files with 18 additions and 5 deletions

View File

@@ -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])