hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-05 03:32:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #9256 from papuSpartan/tomesd

Browse Source 
Integrate optional speed and memory improvements by token merging (via dbolya/tomesd)
This commit is contained in:
AUTOMATIC1111
2023-05-14 08:21:02 +03:00
committed by GitHub
parent 005849331e c2fdb44880
commit 7f6ef764b9
5 changed files with 63 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modules/shared.py
View File

@@ -350,6 +350,8 @@ options_templates.update(options_section(('sd', "Stable Diffusion"), {
"CLIP_stop_at_last_layers": OptionInfo(1, "Clip skip", gr.Slider, {"minimum": 1, "maximum": 12, "step": 1}),
"upcast_attn": OptionInfo(False, "Upcast cross attention layer to float32"),
"randn_source": OptionInfo("GPU", "Random number generator source. Changes seeds drastically. Use CPU to produce the same picture across different vidocard vendors.", gr.Radio, {"choices": ["GPU", "CPU"]}),
"token_merging_ratio_hr": OptionInfo(0, "Merging Ratio (high-res pass)", gr.Slider, {"minimum": 0, "maximum": 0.9, "step": 0.1}),
"token_merging_ratio": OptionInfo(0, "Merging Ratio", gr.Slider, {"minimum": 0, "maximum": 0.9, "step": 0.1})
}))
options_templates.update(options_section(('compatibility', "Compatibility"), {
@@ -458,6 +460,7 @@ options_templates.update(options_section((None, "Hidden options"), {
"sd_checkpoint_hash": OptionInfo("", "SHA256 hash of the current checkpoint"),
}))
options_templates.update()