TabPFN commited on
Commit
96dbc71
1 Parent(s): 6022ee7

Update TabPFN/layer.py

Browse files
Files changed (1) hide show
  1. TabPFN/layer.py +2 -0
TabPFN/layer.py CHANGED
@@ -11,6 +11,8 @@ from torch.nn.modules.transformer import (
11
  Dropout,
12
  LayerNorm,
13
  )
 
 
14
 
15
  from torch.utils.checkpoint import checkpoint
16
 
 
11
  Dropout,
12
  LayerNorm,
13
  )
14
+ import torch
15
+ import torch.nn.functional as F
16
 
17
  from torch.utils.checkpoint import checkpoint
18