diff --git a/modules/initialize.py b/modules/initialize.py
index 74db6722..180e1f8e 100644
--- a/modules/initialize.py
+++ b/modules/initialize.py
@@ -5,6 +5,8 @@ import sys
 import warnings
 import os
 
+from threading import Thread
+
 from modules.timer import startup_timer
 
 
diff --git a/modules/shared_state.py b/modules/shared_state.py
index 61dbdc5b..5da5c7a0 100644
--- a/modules/shared_state.py
+++ b/modules/shared_state.py
@@ -2,6 +2,7 @@ import datetime
 import logging
 import threading
 import time
+import traceback
 import torch
 
 from modules import errors, shared, devices
@@ -159,7 +160,7 @@ class State:
 
             self.current_image_sampling_step = self.sampling_step
 
-        except Exception:
+        except Exception as e:
             # traceback.print_exc()
             # print(e)
             # when switching models during genration, VAE would be on CPU, so creating an image will fail.