minor updates

This commit is contained in:
Vladimir Mandic
2023-11-10 10:04:12 -05:00
parent eda2aa6da2
commit 79af359b1b
4 changed files with 52 additions and 49 deletions
+4 -4
View File
@@ -73,10 +73,10 @@ class DiffusionSampler:
return
for key, value in config.get('All', {}).items(): # apply global defaults
self.config[key] = value
shared.log.debug(f'Sampler: name={name} type=all config={self.config}')
# shared.log.debug(f'Sampler: name={name} type=all config={self.config}')
for key, value in config.get(name, {}).items(): # apply diffusers per-scheduler defaults
self.config[key] = value
shared.log.debug(f'Sampler: name={name} type=scheduler config={self.config}')
# shared.log.debug(f'Sampler: name={name} type=scheduler config={self.config}')
if hasattr(model.scheduler, 'scheduler_config'): # find model defaults
orig_config = model.scheduler.scheduler_config
else:
@@ -84,11 +84,11 @@ class DiffusionSampler:
for key, value in orig_config.items(): # apply model defaults
if key in self.config:
self.config[key] = value
shared.log.debug(f'Sampler: name={name} type=model config={self.config}')
# shared.log.debug(f'Sampler: name={name} type=model config={self.config}')
for key, value in kwargs.items(): # apply user args, if any
if key in self.config:
self.config[key] = value
shared.log.debug(f'Sampler: name={name} type=user config={self.config}')
# shared.log.debug(f'Sampler: name={name} type=user config={self.config}')
# finally apply user preferences
if shared.opts.schedulers_prediction_type != 'default':
self.config['prediction_type'] = shared.opts.schedulers_prediction_type