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

heavily simplify

Browse Source
This commit is contained in:
papuSpartan
2023-05-13 10:23:42 -05:00
parent 55e52c878a
commit ac83627a31
4 changed files with 23 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
modules/sd_models.py
View File

@@ -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
)