Added traceback for bug tracking

Beyondo · Apr 16, 2023 · 666be19 · 666be19
1 parent 572bc1a
commit 666be19
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/legacy/img2img.py b/legacy/img2img.py
@@ -2,7 +2,7 @@
 from legacy import colab, postprocessor, progress
 from IPython.display import display, HTML, Image
 
-import requests
+import requests, traceback
 from PIL import Image
 from io import BytesIO
 importlib.reload(progress)
@@ -49,5 +49,5 @@ def process(ShouldSave, maxNumJobs, ShouldPreview = True, ReplaceResult = True):
             display(HTML("<label>Iterations: %d/%d</label>" % (i + 1,  num_iterations)), display_id="iterations")
         postprocessor.join_queue_thread()
         torch.cuda.empty_cache()
-    except Exception as e:
-        print("Error trying to generate image: " + str(e))
+    except Exception:
+        print("Error trying to generate image", traceback.format_exc())
diff --git a/legacy/text2img.py b/legacy/text2img.py
@@ -1,6 +1,9 @@
 import torch, os, time, datetime, importlib
 from legacy import colab, postprocessor, progress
 from IPython.display import display, HTML
+
+import traceback
+
 importlib.reload(progress)
 importlib.reload(postprocessor)
 def process(ShouldSave, maxNumJobs, ShouldPreview = True, ReplaceResult = True):
@@ -37,5 +40,5 @@ def process(ShouldSave, maxNumJobs, ShouldPreview = True, ReplaceResult = True):
             display(HTML("<label>Iterations: %d/%d</label>" % (i + 1,  num_iterations)), display_id="iterations")
         postprocessor.join_queue_thread()
         torch.cuda.empty_cache()
-    except Exception as e:
-        print("Error trying to generate image: " + str(e))
+    except Exception:
+        print("Error trying to generate image", traceback.format_exc())