This reverts commit e13072cb42
.
This commit is contained in:
parent
388ca351f4
commit
3c32cbb0af
@ -5,6 +5,8 @@ import sys
|
|||||||
import warnings
|
import warnings
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
from threading import Thread
|
||||||
|
|
||||||
from modules.timer import startup_timer
|
from modules.timer import startup_timer
|
||||||
|
|
||||||
|
|
||||||
|
@ -2,6 +2,7 @@ import datetime
|
|||||||
import logging
|
import logging
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
|
import traceback
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
from modules import errors, shared, devices
|
from modules import errors, shared, devices
|
||||||
@ -159,7 +160,7 @@ class State:
|
|||||||
|
|
||||||
self.current_image_sampling_step = self.sampling_step
|
self.current_image_sampling_step = self.sampling_step
|
||||||
|
|
||||||
except Exception:
|
except Exception as e:
|
||||||
# traceback.print_exc()
|
# traceback.print_exc()
|
||||||
# print(e)
|
# print(e)
|
||||||
# when switching models during genration, VAE would be on CPU, so creating an image will fail.
|
# when switching models during genration, VAE would be on CPU, so creating an image will fail.
|
||||||
|
Loading…
Reference in New Issue
Block a user