wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-18 23:55:08 +08:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity

fix: disable SageAttention for Hunyuan3D v2.1 DiT (#12772)

Browse Source
This commit is contained in:
Paulo Muggler Moreira 2026-03-17 03:27:27 +01:00 committed by GitHub
parent 9a870b5102
commit 8cc746a864
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/hunyuan3dv2_1/hunyuandit.py
View File

@ -343,6 +343,7 @@ class CrossAttention(nn.Module):
k.reshape(b, s2, self.num_heads * self.head_dim),
v,
heads=self.num_heads,
low_precision_attention=False,
)
out = self.out_proj(x)
@ -412,6 +413,7 @@ class Attention(nn.Module):
key.reshape(B, N, self.num_heads * self.head_dim),
value,
heads=self.num_heads,
low_precision_attention=False,
)
x = self.out_proj(x)