Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix sampler defaults, facehires strength check

Browse Source
This commit is contained in:
Vladimir Mandic
2024-06-11 08:55:04 -04:00
parent 1fe7d922b8
commit d302ffbfa4
4 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_diffusers.py
+12 -1
View File
@@ -111,13 +111,22 @@ class DiffusionSampler:
return
for key, value in config.get('All', {}).items(): # apply global defaults
self.config[key] = value
debug(f'Sampler: all="{self.config}"')
if hasattr(model.scheduler, 'scheduler_config'): # find model defaults
orig_config = model.scheduler.scheduler_config
else:
orig_config = model.scheduler.config
if not hasattr(model, 'orig_scheduler'): # store settings from initial scheduler
model.orig_scheduler = orig_config.copy()
else:
for key, value in model.orig_scheduler.items(): # apply scheduler defaults
if key in self.config:
self.config[key] = value
debug(f'Sampler: original="{model.orig_scheduler}"')
for key, value in orig_config.items(): # apply model defaults
if key in self.config:
self.config[key] = value
debug(f'Sampler: default="{self.config}"')
for key, value in config.get(name, {}).items(): # apply diffusers per-scheduler defaults
self.config[key] = value
for key, value in kwargs.items(): # apply user args, if any
@@ -168,11 +177,13 @@ class DiffusionSampler:
# validate all config params
signature = inspect.signature(constructor, follow_wrapped=True)
possible = signature.parameters.keys()
debug(f'Sampler: sampler="{name}" config={self.config} signature={possible}')
for key in self.config.copy().keys():
if key not in possible:
shared.log.warning(f'Sampler: sampler="{name}" config={self.config} invalid={key}')
del self.config[key]
debug(f'Sampler: name="{name}"')
debug(f'Sampler: config={self.config}')
debug(f'Sampler: signature={possible}')
# shared.log.debug(f'Sampler: sampler="{name}" config={self.config}')
self.sampler = constructor(**self.config)
# shared.log.debug(f'Sampler: class="{self.sampler.__class__.__name__}" config={self.sampler.config}')