1 files changed, 8 insertions, 0 deletions
diff --git a/modules/sd_samplers_compvis.py b/modules/sd_samplers_compvis.py
index 3d35ff72..88541193 100644
--- a/modules/sd_samplers_compvis.py
+++ b/modules/sd_samplers_compvis.py
@@ -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)