Update configuration_bert2d.py
Browse files- configuration_bert2d.py +3 -1
configuration_bert2d.py
CHANGED
|
@@ -18,7 +18,9 @@
|
|
| 18 |
from collections import OrderedDict
|
| 19 |
from typing import Mapping
|
| 20 |
|
| 21 |
-
from transformers import PretrainedConfig
|
|
|
|
|
|
|
| 22 |
|
| 23 |
logger = logging.get_logger(__name__)
|
| 24 |
|
|
|
|
| 18 |
from collections import OrderedDict
|
| 19 |
from typing import Mapping
|
| 20 |
|
| 21 |
+
from transformers.configuration_utils import PretrainedConfig
|
| 22 |
+
from transformers.onnx import OnnxConfig
|
| 23 |
+
from transformers.utils import logging
|
| 24 |
|
| 25 |
logger = logging.get_logger(__name__)
|
| 26 |
|