Update modeling_dream.py
Browse files- modeling_dream.py +1 -0
modeling_dream.py
CHANGED
|
@@ -25,6 +25,7 @@ import os
|
|
| 25 |
import torch
|
| 26 |
import torch.utils.checkpoint
|
| 27 |
from torch import nn
|
|
|
|
| 28 |
|
| 29 |
from transformers.activations import ACT2FN
|
| 30 |
from transformers.cache_utils import Cache, DynamicCache
|
|
|
|
| 25 |
import torch
|
| 26 |
import torch.utils.checkpoint
|
| 27 |
from torch import nn
|
| 28 |
+
from torch.nn import functional as F
|
| 29 |
|
| 30 |
from transformers.activations import ACT2FN
|
| 31 |
from transformers.cache_utils import Cache, DynamicCache
|