numb3r3 jemfu commited on
Commit
8469b0a
1 Parent(s): 20be9d6

Update block.py (#10)

Browse files

- Update block.py (8ad715efdaab6b1b050e705e81cb338ddce4818e)


Co-authored-by: Jie Fu <[email protected]>

Files changed (1) hide show
  1. block.py +0 -1
block.py CHANGED
@@ -12,7 +12,6 @@ import torch.nn as nn
12
  import torch.nn.functional as F
13
  from torch import Tensor
14
 
15
- from .stochastic_depth import StochasticDepth
16
  from .mha import MHA
17
  from .mlp import Mlp
18
 
 
12
  import torch.nn.functional as F
13
  from torch import Tensor
14
 
 
15
  from .mha import MHA
16
  from .mlp import Mlp
17