Upload modeling_codesage.py with huggingface_hub
Browse files- modeling_codesage.py +2 -1
modeling_codesage.py
CHANGED
|
@@ -8,7 +8,8 @@ import torch.utils.checkpoint
|
|
| 8 |
from torch import nn
|
| 9 |
from torch.nn import CrossEntropyLoss, MSELoss, BCEWithLogitsLoss
|
| 10 |
from transformers.activations import ACT2FN
|
| 11 |
-
from transformers.modeling_utils import Conv1D, PreTrainedModel
|
|
|
|
| 12 |
from transformers.utils import logging
|
| 13 |
from .config_codesage import CodeSageConfig
|
| 14 |
from transformers.modeling_outputs import (
|
|
|
|
| 8 |
from torch import nn
|
| 9 |
from torch.nn import CrossEntropyLoss, MSELoss, BCEWithLogitsLoss
|
| 10 |
from transformers.activations import ACT2FN
|
| 11 |
+
# from transformers.modeling_utils import Conv1D, PreTrainedModel
|
| 12 |
+
from transformers import Conv1D, PreTrainedModel
|
| 13 |
from transformers.utils import logging
|
| 14 |
from .config_codesage import CodeSageConfig
|
| 15 |
from transformers.modeling_outputs import (
|