Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit 672915a

Browse files
committed
remove extra arg.
1 parent 7ffec73 commit 672915a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchchat/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,9 +61,9 @@ class MultiModalProjector(nn.Module):
6161
def __init__(self, in_channels: int, out_channels: int, act: nn.Module):
6262
super().__init__()
6363

64-
self.linear_1 = nn.Linear(args.in_channels, args.out_channels, bias=True)
64+
self.linear_1 = nn.Linear(in_channels, out_channels, bias=True)
6565
self.act = act
66-
self.linear_2 = nn.Linear(args.out_channels, args.out_channels, bias=True)
66+
self.linear_2 = nn.Linear(out_channels, out_channels, bias=True)
6767

6868
def forward(self, image_features):
6969
hidden_states = self.linear_1(image_features)

0 commit comments

Comments
 (0)