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

paulomuggler · web-flow · commit 8cc746a86411 · 2026-03-16T22:27:27.000-04:00
diff --git a/comfy/ldm/hunyuan3dv2_1/hunyuandit.py b/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
@@ -343,6 +343,7 @@ def forward(self, x, y):
             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 @@ def forward(self, x):
             key.reshape(B, N, self.num_heads * self.head_dim),
             value,
             heads=self.num_heads,
+            low_precision_attention=False,
         )
 
         x = self.out_proj(x)

Original file line number	Diff line number	Diff line change
`@@ -343,6 +343,7 @@ def forward(self, x, y):`
`343`	`343`	`k.reshape(b, s2, self.num_heads * self.head_dim),`
`344`	`344`	`v,`
`345`	`345`	`heads=self.num_heads,`
	`346`	`+ low_precision_attention=False,`
`346`	`347`	`)`
`347`	`348`
`348`	`349`	`out = self.out_proj(x)`
`@@ -412,6 +413,7 @@ def forward(self, x):`
`412`	`413`	`key.reshape(B, N, self.num_heads * self.head_dim),`
`413`	`414`	`value,`
`414`	`415`	`heads=self.num_heads,`
	`416`	`+ low_precision_attention=False,`
`415`	`417`	`)`
`416`	`418`
`417`	`419`	`x = self.out_proj(x)`