add ip adapter and hdr latent control

This commit is contained in:
Vladimir Mandic
2023-11-24 18:06:09 -05:00
parent 1d5eb29550
commit bcba5cadab
8 changed files with 197 additions and 3 deletions
+3 -1
View File
@@ -17,6 +17,7 @@ import modules.errors as errors
from modules.processing import StableDiffusionProcessing, create_random_tensors
import modules.prompt_parser_diffusers as prompt_parser_diffusers
from modules.sd_hijack_hypertile import hypertile_set
from modules.processing_correction import correction_callback
def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_prompts):
@@ -71,7 +72,8 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
raise AssertionError('Interrupted...')
time.sleep(0.1)
def diffusers_callback(_pipe, step: int, _timestep: int, kwargs: dict):
def diffusers_callback(_pipe, step: int, timestep: int, kwargs: dict):
latents = correction_callback(p, timestep, kwargs)
latents = kwargs['latents']
shared.state.sampling_step = step
shared.state.current_latent = latents