Signed-off-by: Vladimir Mandic <mandic00@live.com>
pull/4356/head
Vladimir Mandic 2025-11-03 19:37:31 -05:00
parent 05261e708a
commit 7c1a9a4b9b
2 changed files with 1 additions and 2 deletions

View File

@ -138,7 +138,7 @@ def get_model(model_type = 'decoder', variant = None):
def decode(latents):
global first_run
global first_run # pylint: disable=global-statement
with lock:
vae, variant = get_model(model_type='decoder')
if vae is None or max(latents.shape) > 256: # safetey check of large tensors

View File

@ -1,4 +1,3 @@
import sys
import diffusers
import transformers
from modules import shared, devices, sd_models, model_quant, sd_hijack_te, sd_hijack_vae