wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-08 23:47:24 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Merge ddb3bcc5c0 into b08debceca

Browse Source
This commit is contained in:
Peter Willemsen 2026-07-06 17:34:00 +08:00 committed by GitHub
commit 9ed21e83dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/text_encoders/qwen35.py
View File

@ -451,6 +451,14 @@ class Qwen35VisionPatchEmbed(nn.Module):
def forward(self, x):
x = x.view(-1, self.in_channels, self.temporal_patch_size, self.patch_size, self.patch_size)
if (
comfy.model_management.is_amd()
and x.is_cuda
and x.dtype in (torch.float16, torch.bfloat16)
):
# This Conv3d is a full-patch projection, equivalent to Linear.
# Avoid the ROCm/MIOpen reduced-precision Conv3d kernel that can segfault.
return F.linear(x.flatten(1), self.proj.weight.flatten(1), self.proj.bias)
return self.proj(x).view(-1, self.embed_dim)