From 0b2612133595c4ad020a8c93ee075cb2edb700a0 Mon Sep 17 00:00:00 2001
From: drhead <1313496+drhead@users.noreply.github.com>
Date: Sat, 3 May 2025 14:39:03 -0400
Subject: [PATCH] re-add line to update sampling step (#2853)

---
 modules/shared_state.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/shared_state.py b/modules/shared_state.py
index 699cb0fc..6ea8e961 100644
--- a/modules/shared_state.py
+++ b/modules/shared_state.py
@@ -172,6 +172,8 @@ class State:
                 else:
                     self.assign_current_image(modules.sd_samplers.sample_to_image(self.current_latent))
 
+            self.current_image_sampling_step = self.sampling_step
+
         except Exception as e:
             # traceback.print_exc()
             # print(e)