Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-29 08:31:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

extra network tags

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-01 11:51:22 -04:00
parent 106075e609
commit 1b2890d02c
11 changed files with 191 additions and 164 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+7
View File
@@ -1,3 +1,4 @@
import time
import inspect
import typing
import torch
@@ -54,6 +55,12 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
shared.state.current_latent = latents
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
if shared.state.paused:
shared.log.debug('Sampling paused')
while shared.state.paused:
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
time.sleep(0.1)
def full_vae_decode(latents, model):
shared.log.debug(f'VAE decode: name={sd_vae.loaded_vae_file if sd_vae.loaded_vae_file is not None else "baked"} dtype={model.vae.dtype} upcast={model.vae.config.get("force_upcast", None)} images={latents.shape[0]}')