Compare commits

...

7 Commits

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)