Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

rework diffusers args as and introduce second pass as replacement for hires

Browse Source
This commit is contained in:
Vladimir Mandic
2023-07-14 15:00:43 -04:00
parent 14d1136fe7
commit 8b24efe1b5
11 changed files with 128 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_diffusers.py
+3
View File
@@ -14,6 +14,7 @@ try:
# KDPM2DiscreteScheduler,
PNDMScheduler,
UniPCMultistepScheduler,
LMSDiscreteScheduler,
)
except Exception as e:
import diffusers
@@ -32,6 +33,7 @@ config = {
'PNDM': { 'skip_prk_steps': False, 'set_alpha_to_one': False, 'steps_offset': 0 },
'DPM 1S': { 'solver_order': 2, 'thresholding': False, 'sample_max_value': 1.0, 'algorithm_type': "dpmsolver++", 'solver_type': "midpoint", 'lower_order_final': True, 'use_karras_sigmas': False },
'DPM 2M': { 'thresholding': False, 'sample_max_value': 1.0, 'algorithm_type': "dpmsolver++", 'solver_type': "midpoint", 'lower_order_final': True, 'use_karras_sigmas': False },
'LMSD': { 'use_karras_sigmas': False, 'timestep_spacing': 'linspace', 'steps_offset': 0 },
}
samplers_data_diffusers = [
@@ -45,6 +47,7 @@ samplers_data_diffusers = [
sd_samplers_common.SamplerData('Euler a', lambda model: DiffusionSampler('Euler a', EulerAncestralDiscreteScheduler, model), [], {}),
sd_samplers_common.SamplerData('Heun', lambda model: DiffusionSampler('Heun', HeunDiscreteScheduler, model), [], {}),
sd_samplers_common.SamplerData('PNDM', lambda model: DiffusionSampler('PNDM', PNDMScheduler, model), [], {}),
sd_samplers_common.SamplerData('LMSD', lambda model: DiffusionSampler('LMSD', LMSDiscreteScheduler, model), [], {}),
]
class DiffusionSampler: