CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
Haoming 2026-02-21 15:15:06 +08:00
parent cd0c67cc5a
commit f043f2c3e9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extensions-builtin/sd_forge_radial/scripts/forge_radial.py
View File

@ -9,7 +9,7 @@ if TYPE_CHECKING:
import gradio as gr
import torch
import torch.nn.functional as F
from lib_radial.attn_mask import RADIAL_ENABLE, MaskMap, RadialAttention
from lib_radial.attn_mask import RADIAL_ENABLE
from backend.nn import wan
from modules import scripts
@ -19,6 +19,8 @@ ORIG_ATTENTION = wan.attention
def get_radial_attn_func(video_token_num, num_frame, block_size, decay_factor, allow_compile):
from lib_radial.attn_mask import MaskMap, RadialAttention
mask_map = MaskMap(video_token_num, num_frame)
def radial_attn_func(q, k, v, heads, mask=None, attn_precision=None, skip_reshape=False, skip_output_reshape=False, **kwargs):