Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-16 20:23:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed auto switch attention

Browse Source
This commit is contained in:
deepbeepmeep 2025-08-09 12:25:20 +02:00 committed by GitHub
parent b8d843e7db
commit 8039080ea4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
wgp.py
View File

@ -3988,11 +3988,9 @@ def generate_video(
overridden_attention = get_overridden_attention(model_type)
# if overridden_attention is not None and overridden_attention != attention_mode: print(f"Attention mode has been overriden to {overridden_attention} for model type '{model_type}'")
attn = overridden_attention if overridden_attention is not None else attention_mode
if attention_mode == "auto":
if attn == "auto":
attn = get_auto_attention()
elif attention_mode in attention_modes_supported:
attn = attention_mode
else:
elif not attn in attention_modes_supported:
send_cmd("info", f"You have selected attention mode '{attention_mode}'. However it is not installed or supported on your system. You should either install it or switch to the default 'sdpa' attention.")
send_cmd("exit")
return