lhallee commited on
Commit
cf10afe
·
verified ·
1 Parent(s): 01b8c88

Upload modeling_esm_plusplus.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- from embedding_mixin import EmbeddingMixin, Pooler
 
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