mirror of https://github.com/vladmandic/automatic
parent
05261e708a
commit
7c1a9a4b9b
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue