Merge pull request #1188 from cheald/fix-preview-error

Fix errors on early termination
Sygil-Dev · Sep 16, 2022 · 47e39df · 47e39df
2 parents ac99b65 + 419f3f9
commit 47e39df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/sd_utils.py b/scripts/sd_utils.py
@@ -787,6 +787,8 @@ def ModelLoader(models,load=False,unload=False,imgproc_realesrgan_model_name='Re
 #
 @retry(tries=5)
 def generation_callback(img, i=0):
+    if "update_preview_frequency" not in st.session_state:
+        return
 
     try:
         if i == 0: