lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update controlnet.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 14:41:10 -08:00
parent 08763a234d
commit ec7adb41fa
1 changed files with 9 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@@ -1005,40 +1005,20 @@ class Script(scripts.Script, metaclass=(
self.detected_map = detected_maps self.detected_map = detected_maps
self.post_processors = post_processors self.post_processors = post_processors
def controlnet_hack(self, p): def process_unit_after_click_generate(self, p, unit, *args, **kwargs):
t = time.time() return
if getattr(shared.cmd_opts, 'controlnet_tracemalloc', False):
tracemalloc.start()
setattr(self, "malloc_begin", tracemalloc.take_snapshot())
self.controlnet_main_entry(p) def process_unit_before_every_sampling(self, p, unit, *args, **kwargs):
if getattr(shared.cmd_opts, 'controlnet_tracemalloc', False): return
logger.info("After hook malloc:")
for stat in tracemalloc.take_snapshot().compare_to(self.malloc_begin, "lineno")[:10]:
logger.info(stat)
if len(self.enabled_units) > 0:
logger.info(f'ControlNet Patched - Time = {time.time() - t}')
@staticmethod
def process_has_sdxl_refiner(p):
return getattr(p, 'refiner_checkpoint', None) is not None
def process(self, p, *args, **kwargs): def process(self, p, *args, **kwargs):
if not Script.process_has_sdxl_refiner(p): for unit in Script.get_enabled_units(p):
self.controlnet_hack(p) self.process_unit_after_click_generate(p, unit, *args, **kwargs)
return return
def before_process_batch(self, p, *args, **kwargs): def process_before_every_sampling(self, p, *args, **kwargs):
if Script.process_has_sdxl_refiner(p): for unit in Script.get_enabled_units(p):
self.controlnet_hack(p) self.process_unit_before_every_sampling(p, unit, *args, **kwargs)
return
def postprocess_batch(self, p, *args, **kwargs):
images = kwargs.get('images', [])
for post_processor in self.post_processors:
for i in range(len(images)):
images[i] = post_processor(images[i])
return return
def postprocess(self, p, processed, *args): def postprocess(self, p, processed, *args):