comi-logic/Wan2GP
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added Preview mode and support Sky Reels v2 Diffusion Forcing

Browse Source
This commit is contained in:
deepbeepmeep
2025-04-25 22:07:13 +02:00
parent a63aff0377
commit 7c447ea36c
8 changed files with 1155 additions and 832 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
wan/modules/sage2_core.py
View File

@@ -1075,13 +1075,13 @@ def sageattn_qk_int8_pv_fp8_cuda_sm90(
q_int8, q_scale, k_int8, k_scale = per_thread_int8_triton(q, k, km, tensor_layout=tensor_layout, BLKQ=64, WARPQ=16, BLKK=128, WARPK=128)
q_size = q.size()
kv_len = k.size(seq_dim)
q_device = q.device
del q,k
# pad v to multiple of 128
# TODO: modify per_channel_fp8 kernel to handle this
kv_len = k.size(seq_dim)
v_pad_len = 128 - (kv_len % 128) if kv_len % 128 != 0 else 0
if v_pad_len > 0:
if tensor_layout == "HND":