"vscode:/vscode.git/clone" did not exist on "3597b06a4faafae96f9696e736ebc198520dc8ec"
Commit 71e4b817 authored by zhuwenwen's avatar zhuwenwen
Browse files

update conv layout

parent 54b9592c
......@@ -436,7 +436,7 @@ class Qwen2_5_VisionPatchEmbed(nn.Module):
L, C = x.shape
x = x.view(L, -1, self.temporal_patch_size, self.patch_size,
self.patch_size)
# x=x.to(memory_format=torch.channels_last_3d)
x = x.to(memory_format=torch.channels_last_3d)
x = self.proj(x).view(L, self.hidden_size)
return x
......
......@@ -464,7 +464,7 @@ class Qwen2VisionPatchEmbed(nn.Module):
L, C = x.shape
x = x.view(L, -1, self.temporal_patch_size, self.patch_size,
self.patch_size)
# x=x.to(memory_format=torch.channels_last_3d)
x = x.to(memory_format=torch.channels_last_3d)
x = self.proj(x).view(L, self.embed_dim)
return x
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment