daniel/h3-blackwell-runtime
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix F.conv3d call: remove invalid kernel_size kwarg (uses weight.shape)

Browse source
This commit is contained in:
Daniel Maddern 2026-08-19 22:01:12 +07:00
parent 453aa86328
commit 6866263631
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/h3_blackwell_runtime/qwen3vl_vision.py
View file

@ -213,7 +213,8 @@ class _VisionPatchEmbed(nn.Module):
def forward(self, x: torch.Tensor) -> torch.Tensor: def forward(self, x: torch.Tensor) -> torch.Tensor:
target = self.weight.dtype target = self.weight.dtype
x = x.view(-1, 3, VISION_TEMPORAL, VISION_PATCH, VISION_PATCH) x = x.view(-1, 3, VISION_TEMPORAL, VISION_PATCH, VISION_PATCH)
return F.conv3d(x.to(target), self.weight, self.bias, kernel_size=(VISION_TEMPORAL, VISION_PATCH, VISION_PATCH), stride=(VISION_TEMPORAL, VISION_PATCH, VISION_PATCH)).view(-1, self.weight.shape[0]) s = (VISION_TEMPORAL, VISION_PATCH, VISION_PATCH)
return F.conv3d(x.to(target), self.weight, self.bias, stride=s).view(-1, self.weight.shape[0])
class _VisionMLP(nn.Module): class _VisionMLP(nn.Module):