PommesPeter commited on
Commit
0b9e35d
1 Parent(s): 7a92cca

Update models/model.py

Browse files
Files changed (1) hide show
  1. models/model.py +1 -1
models/model.py CHANGED
@@ -543,7 +543,7 @@ class TransformerBlock(nn.Module):
543
  nn.SiLU(),
544
  nn.Linear(
545
  min(dim, 1024),
546
- 2 * dim,
547
  bias=True,
548
  ),
549
  )
 
543
  nn.SiLU(),
544
  nn.Linear(
545
  min(dim, 1024),
546
+ 4 * dim,
547
  bias=True,
548
  ),
549
  )