aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_models.py
diff options
context:
space:
mode:
authorpapuSpartan <30642826+papuSpartan@users.noreply.github.com>2023-05-13 15:23:42 +0000
committerpapuSpartan <30642826+papuSpartan@users.noreply.github.com>2023-05-13 15:23:42 +0000
commitac83627a31daac06f4d48b0e7db223ef807fe8e5 (patch)
tree68d89cf786fa340f87eb7b8c1afe7ccae463ccd2 /modules/sd_models.py
parent55e52c878ab669d5b11b001a4152ee1a3b8d4880 (diff)
downloadstable-diffusion-webui-gfx803-ac83627a31daac06f4d48b0e7db223ef807fe8e5.tar.gz
stable-diffusion-webui-gfx803-ac83627a31daac06f4d48b0e7db223ef807fe8e5.tar.bz2
stable-diffusion-webui-gfx803-ac83627a31daac06f4d48b0e7db223ef807fe8e5.zip
heavily simplify
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r--modules/sd_models.py11
1 files changed, 4 insertions, 7 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 4787193c..4c9a0a1f 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -596,11 +596,8 @@ def apply_token_merging(sd_model, hr: bool):
tomesd.apply_patch(
sd_model,
ratio=ratio,
- max_downsample=shared.opts.token_merging_maximum_down_sampling,
- sx=shared.opts.token_merging_stride_x,
- sy=shared.opts.token_merging_stride_y,
- use_rand=shared.opts.token_merging_random,
- merge_attn=shared.opts.token_merging_merge_attention,
- merge_crossattn=shared.opts.token_merging_merge_cross_attention,
- merge_mlp=shared.opts.token_merging_merge_mlp
+ use_rand=False, # can cause issues with some samplers
+ merge_attn=True,
+ merge_crossattn=False,
+ merge_mlp=False
)