model component merge

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-01-19 14:36:09 -05:00
parent de1063275c
commit 311d402b0c
11 changed files with 833 additions and 245 deletions
+7 -10
View File
@@ -15,26 +15,20 @@ try:
CMStochasticIterativeScheduler,
UniPCMultistepScheduler,
DDIMScheduler,
EulerDiscreteScheduler,
EulerAncestralDiscreteScheduler,
EDMEulerScheduler,
FlowMatchEulerDiscreteScheduler,
DEISMultistepScheduler,
SASolverScheduler,
DPMSolverSinglestepScheduler,
DPMSolverMultistepScheduler,
EDMDPMSolverMultistepScheduler,
CosineDPMSolverMultistepScheduler,
DPMSolverSDEScheduler,
HeunDiscreteScheduler,
FlowMatchHeunDiscreteScheduler,
LCMScheduler,
PNDMScheduler,
IPNDMScheduler,
DDPMScheduler,
@@ -172,14 +166,17 @@ class DiffusionSampler:
return
self.name = name
self.config = {}
if not hasattr(model, 'scheduler'):
return
if getattr(model, "default_scheduler", None) is None: # sanity check
self.sampler = None
# if not hasattr(model, 'scheduler'):
# return
if getattr(model, "default_scheduler", None) is None and (model is not None): # sanity check
model.default_scheduler = copy.deepcopy(model.scheduler)
for key, value in config.get('All', {}).items(): # apply global defaults
self.config[key] = value
debug_log(f'Sampler: all="{self.config}"')
if hasattr(model.default_scheduler, 'scheduler_config'): # find model defaults
if model is None:
orig_config = {}
elif hasattr(model.default_scheduler, 'scheduler_config'): # find model defaults
orig_config = model.default_scheduler.scheduler_config
else:
orig_config = model.default_scheduler.config