Update ultimate-upscale.py

Fixed image saving duplicating the prompt by number of tiles and using the grid format rather than image format
This commit is contained in:
Emmortal451
2023-04-30 14:54:53 -04:00
parent 0a3d03a41a
commit 59515c0ff1

View File

@@ -97,7 +97,10 @@ class USDUpscaler():
self.seams_fix.enabled = self.seams_fix.mode != USDUSFMode.NONE
def save_image(self):
images.save_image(self.image, self.p.outpath_samples, "", self.p.seed, self.p.prompt, opts.grid_format, info=self.initial_info, p=self.p)
if type(self.p.prompt) != list:
images.save_image(self.image, self.p.outpath_samples, "", self.p.seed, self.p.prompt, opts.samples_format, info=self.initial_info, p=self.p)
else:
images.save_image(self.image, self.p.outpath_samples, "", self.p.seed, self.p.prompt[0], opts.samples_format, info=self.initial_info, p=self.p)
def calc_jobs_count(self):
redraw_job_count = (self.rows * self.cols) if self.redraw.enabled else 0