aboutsummaryrefslogtreecommitdiff
path: root/modules/sd_samplers_timesteps.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-08-08 22:27:32 +0300
committerAUTOMATIC1111 <16777216c@gmail.com>2023-08-08 22:27:32 +0300
commit0e83c675257f473e024511845e7940802333fd5f (patch)
tree57f3b4e3f7238a1ccce4cc52eae953c4ae31e9cb /modules/sd_samplers_timesteps.py
parent70c63c1208d33bf02e15c4e310bac83f12ee8625 (diff)
by request: fix tiled vae extension
Diffstat (limited to 'modules/sd_samplers_timesteps.py')
-rw-r--r--modules/sd_samplers_timesteps.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/sd_samplers_timesteps.py b/modules/sd_samplers_timesteps.py
index d89d0efb..f61799a8 100644
--- a/modules/sd_samplers_timesteps.py
+++ b/modules/sd_samplers_timesteps.py
@@ -1,5 +1,6 @@
import torch
import inspect
+import sys
from modules import devices, sd_samplers_common, sd_samplers_timesteps_impl
from modules.sd_samplers_cfg_denoiser import CFGDenoiser
@@ -145,3 +146,6 @@ class CompVisSampler(sd_samplers_common.Sampler):
return samples
+
+sys.modules['modules.sd_samplers_compvis'] = sys.modules[__name__]
+VanillaStableDiffusionSampler = CompVisSampler # temp. compatibility with older extensions