From 3c32cbb0afcf5a0f0a06283fedcffbb318d59c8a Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Fri, 9 Feb 2024 02:34:58 +0000
Subject: [PATCH] Revert "Fix more ruff lint (#139)" (#144)

This reverts commit e13072cb42813a951ac81f7c14c02986c8cdeae0.
---
 modules/initialize.py   | 2 ++
 modules/shared_state.py | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

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.