Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

diffusers better ti reload

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-26 17:40:29 -04:00
parent 2f17cef6f6
commit 6bb1e59de6
6 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_compvis.py
+3 -3
View File
@@ -7,11 +7,11 @@ import torch
from modules.shared import state
from modules import sd_samplers_common, prompt_parser, shared
import modules.uni_pc
import modules.unipc
samplers_data_compvis = [
sd_samplers_common.SamplerData('UniPC', lambda model: VanillaStableDiffusionSampler(modules.uni_pc.UniPCSampler, model), [], {}),
sd_samplers_common.SamplerData('UniPC', lambda model: VanillaStableDiffusionSampler(modules.unipc.UniPCSampler, model), [], {}),
sd_samplers_common.SamplerData('DDIM', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.ddim.DDIMSampler, model), [], {"default_eta_is_0": True}),
sd_samplers_common.SamplerData('PLMS', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.plms.PLMSSampler, model), [], {}),
]
@@ -22,7 +22,7 @@ class VanillaStableDiffusionSampler:
self.sampler = constructor(sd_model)
self.is_ddim = hasattr(self.sampler, 'p_sample_ddim')
self.is_plms = hasattr(self.sampler, 'p_sample_plms')
self.is_unipc = isinstance(self.sampler, modules.uni_pc.UniPCSampler)
self.is_unipc = isinstance(self.sampler, modules.unipc.UniPCSampler)
self.orig_p_sample_ddim = None
if self.is_plms:
self.orig_p_sample_ddim = self.sampler.p_sample_plms