Abdullah-Nazhat commited on
Commit
d3cadd1
·
verified ·
1 Parent(s): 5c2da1c

Update interactor.py

Browse files
Files changed (1) hide show
  1. interactor.py +3 -3
interactor.py CHANGED
@@ -11,9 +11,9 @@ class MappingUnit(nn.Module):
11
 
12
 
13
  self.norm_token = nn.LayerNorm(dim)
14
- self.proj_1 = nn.Linear(dim,dim)
15
- self.proj_2 = nn.Linear(dim,dim)
16
- self.proj_3 = nn.Linear(dim,dim)
17
  self.gelu = nn.GELU()
18
 
19
  def forward(self, x):
 
11
 
12
 
13
  self.norm_token = nn.LayerNorm(dim)
14
+ self.proj_1 = nn.Linear(dim,dim,bias = False)
15
+ self.proj_2 = nn.Linear(dim,dim,bias = False)
16
+ self.proj_3 = nn.Linear(dim,dim,bias = False)
17
  self.gelu = nn.GELU()
18
 
19
  def forward(self, x):