Upload modeling_esm_plusplus.py with huggingface_hub
Browse files- modeling_esm_plusplus.py +2 -1
modeling_esm_plusplus.py
CHANGED
|
@@ -29,7 +29,8 @@ from torch.utils.data import DataLoader
|
|
| 29 |
from tqdm.auto import tqdm
|
| 30 |
from transformers import PreTrainedModel, PreTrainedTokenizerFast, PreTrainedTokenizerBase, PretrainedConfig
|
| 31 |
from transformers.modeling_outputs import ModelOutput
|
| 32 |
-
|
|
|
|
| 33 |
|
| 34 |
try:
|
| 35 |
from torch.nn.attention.flex_attention import create_block_mask
|
|
|
|
| 29 |
from tqdm.auto import tqdm
|
| 30 |
from transformers import PreTrainedModel, PreTrainedTokenizerFast, PreTrainedTokenizerBase, PretrainedConfig
|
| 31 |
from transformers.modeling_outputs import ModelOutput
|
| 32 |
+
|
| 33 |
+
from .embedding_mixin import EmbeddingMixin, Pooler
|
| 34 |
|
| 35 |
try:
|
| 36 |
from torch.nn.attention.flex_attention import create_block_mask
|