From 549c02a874eac1b3ac78ec61b7f1beec81dd76aa Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Sat, 3 Jun 2023 10:36:53 -0400
Subject: [PATCH] minor fixes

---
 javascript/progressbar.js | 1 -
 modules/images.py         | 1 -
 modules/processing.py     | 2 +-
 scripts/xyz_grid.py       | 1 +
 4 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/javascript/progressbar.js b/javascript/progressbar.js
index 82ac1276d..24e7f50ac 100644
--- a/javascript/progressbar.js
+++ b/javascript/progressbar.js
@@ -45,7 +45,6 @@ function checkPaused(state) {
 }
 
 function setProgress(res) {
-  console.log(res);
   elements = ['txt2img_generate', 'img2img_generate', 'extras_generate']
   const progress = (res?.progress || 0)
   const perc = res && (progress > 0) ? `${Math.round(100.0 * progress)}%` : ''
diff --git a/modules/images.py b/modules/images.py
index d068842eb..db812eb38 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -434,7 +434,6 @@ def atomically_save_image():
         filename = filename.strip()
         if extension[0] != '.': # add dot if missing
             extension = '.' + extension
-        print('image_format', extension)
         try:
             image_format = Image.registered_extensions()[extension]
         except Exception:
diff --git a/modules/processing.py b/modules/processing.py
index 003ca69bc..8e4ad13d2 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -922,7 +922,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
             decoded_samples = torch.from_numpy(np.array(batch_images))
             decoded_samples = decoded_samples.to(shared.device)
             decoded_samples = 2. * decoded_samples - 1.
-            if shared.opts.sd_vae_sliced_encode:
+            if shared.opts.sd_vae_sliced_encode and len(decoded_samples) > 1:
                 samples = torch.stack([
                     self.sd_model.get_first_stage_encoding(
                         self.sd_model.encode_first_stage(torch.unsqueeze(decoded_sample, 0))
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
index 041698ee5..cfa56fa12 100644
--- a/scripts/xyz_grid.py
+++ b/scripts/xyz_grid.py
@@ -73,6 +73,7 @@ def apply_checkpoint(p, x, xs):
         shared.log.warning(f"XYZ grid: unknown checkpoint: {x}")
     else:
         sd_models.reload_model_weights(shared.sd_model, info)
+        p.override_settings['sd_model_checkpoint'] = info.name
 
 
 def confirm_checkpoints(p, xs):