hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

split samplers into one more files for k-diffusion

Browse Source
This commit is contained in:
AUTOMATIC
2023-01-30 10:11:30 +03:00
parent 274474105a
commit 4df63d2d19
4 changed files with 22 additions and 348 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules/sd_samplers_compvis.py
View File

@@ -1,4 +1,6 @@
import math
import ldm.models.diffusion.ddim
import ldm.models.diffusion.plms
import numpy as np
import torch
@@ -7,6 +9,12 @@ from modules.shared import state
from modules import sd_samplers_common, prompt_parser, shared
samplers_data_compvis = [
sd_samplers_common.SamplerData('DDIM', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.ddim.DDIMSampler, model), [], {}),
sd_samplers_common.SamplerData('PLMS', lambda model: VanillaStableDiffusionSampler(ldm.models.diffusion.plms.PLMSSampler, model), [], {}),
]
class VanillaStableDiffusionSampler:
def __init__(self, constructor, sd_model):
self.sampler = constructor(sd_model)