mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
model component merge
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user