Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 09:38:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

rename wan a14b stages

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-18 16:10:32 -04:00
parent 8f151b7e17
commit 13195dadb9
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared.py
+1 -1
View File
@@ -160,7 +160,7 @@ options_templates.update(options_section(('model_options', "Models Options"), {
"model_h1_sep": OptionInfo("<h2>HiDream</h2>", "", gr.HTML),
"model_h1_llama_repo": OptionInfo("Default", "LLama repo", gr.Textbox),
"model_wan_sep": OptionInfo("<h2>WanAI</h2>", "", gr.HTML),
"model_wan_stage": OptionInfo("first", "Processing stage", gr.Radio, {"choices": ['first', 'second', 'both'] }),
"model_wan_stage": OptionInfo("first", "Processing stage", gr.Radio, {"choices": ['high noise', 'low noise', 'combined'] }),
"model_wan_boundary": OptionInfo(0.85, "Stage boundary ratio", gr.Slider, {"minimum": 0, "maximum": 1.0, "step": 0.05 }),
}))
pipelines/model_wanai.py
+3 -3
View File
@@ -60,13 +60,13 @@ def load_wan(checkpoint_info, diffusers_load_config={}):
sd_models.hf_auth_check(checkpoint_info)
if 'a14b' in repo_id.lower():
if shared.opts.model_wan_stage == 'first':
if shared.opts.model_wan_stage == 'high noise':
transformer = load_transformer(repo_id, diffusers_load_config, 'transformer')
transformer_2 = None
elif shared.opts.model_wan_stage == 'second':
elif shared.opts.model_wan_stage == 'low noise':
transformer = load_transformer(repo_id, diffusers_load_config, 'transformer_2')
transformer_2 = None
elif shared.opts.model_wan_stage == 'both':
elif shared.opts.model_wan_stage == 'combined':
transformer = load_transformer(repo_id, diffusers_load_config, 'transformer')
transformer_2 = load_transformer(repo_id, diffusers_load_config, 'transformer_2')
else: