experimental sampler monkeypatch

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-02 07:06:34 -04:00
parent f7e433bd07
commit 461bed5720
3 changed files with 9 additions and 3 deletions
+6
View File
@@ -317,6 +317,12 @@ class DiffusionSampler:
self.sampler = None
return
# monkey-patch to allow sdxl pipeline to execute flowmatch samplers
if not hasattr(sampler, 'scale_model_input'):
sampler.scale_model_input = lambda x, _y: x
if not hasattr(sampler, 'init_noise_sigma'):
sampler.init_noise_sigma = 1.0
self.sampler = sampler
# shared.log.debug_log(f'Sampler: class="{self.sampler.__class__.__name__}" config={self.sampler.config}')