From 59515c0ff1715d797a4d68d62f2f9c1baae15059 Mon Sep 17 00:00:00 2001
From: Emmortal451 <45298380+Emmortal451@users.noreply.github.com>
Date: Sun, 30 Apr 2023 14:54:53 -0400
Subject: [PATCH] Update ultimate-upscale.py

Fixed image saving duplicating the prompt by number of tiles and using the grid format rather than image format
---
 scripts/ultimate-upscale.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/scripts/ultimate-upscale.py b/scripts/ultimate-upscale.py
index 458ea05..7a0cc2a 100644
--- a/scripts/ultimate-upscale.py
+++ b/scripts/ultimate-upscale.py
@@ -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