Spaces:
Runtime error
Runtime error
Update TabPFN/layer.py
Browse files- 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 |
|