mirror of
https://github.com/vladmandic/automatic
synced 2026-09-05 04:20:44 +02:00
@@ -90,11 +90,10 @@ def full_vae_decode(latents, model):
|
||||
if debug:
|
||||
devices.torch_gc(force=True)
|
||||
shared.mem_mon.reset()
|
||||
|
||||
if shared.opts.diffusers_offload_mode != "sequential":
|
||||
sd_models.move_model(model.vae, devices.device)
|
||||
|
||||
sd_models.set_vae_options(model, vae=None, op='decode')
|
||||
# sd_models.set_vae_options(model, vae=None, op='decode') # set during model load
|
||||
upcast = (model.vae.dtype == torch.float16) and (getattr(model.vae.config, 'force_upcast', False) or shared.opts.no_half_vae)
|
||||
if upcast:
|
||||
if hasattr(model, 'upcast_vae'): # this is done by diffusers automatically if output_type != 'latent'
|
||||
|
||||
Reference in New Issue
Block a user