robinzixuan
commited on
Commit
•
8b594fd
1
Parent(s):
3bad039
Upload modeling_bert.py
Browse files- modeling_bert.py +1 -1
modeling_bert.py
CHANGED
@@ -20,7 +20,7 @@ import os
|
|
20 |
import warnings
|
21 |
from dataclasses import dataclass
|
22 |
from typing import List, Optional, Tuple, Union
|
23 |
-
|
24 |
import torch
|
25 |
import torch.utils.checkpoint
|
26 |
from packaging import version
|
|
|
20 |
import warnings
|
21 |
from dataclasses import dataclass
|
22 |
from typing import List, Optional, Tuple, Union
|
23 |
+
from functools import partial
|
24 |
import torch
|
25 |
import torch.utils.checkpoint
|
26 |
from packaging import version
|