isoformer-anonymous commited on
Commit
13a7e78
1 Parent(s): 0e799f0

Update modeling_isoformer.py

Browse files
Files changed (1) hide show
  1. modeling_isoformer.py +4 -4
modeling_isoformer.py CHANGED
@@ -1,8 +1,8 @@
1
- from isoformer_config import IsoformerConfig
2
  from transformers import PreTrainedModel
3
- from modeling_esm import NTForMaskedLM, MultiHeadAttention
4
- from esm_config import NTConfig
5
- from modeling_esm_original import EsmForMaskedLM
6
  from transformers.models.esm.configuration_esm import EsmConfig
7
  from enformer_pytorch import Enformer, str_to_one_hot, EnformerConfig
8
  import torch
 
1
+ from .isoformer_config import IsoformerConfig
2
  from transformers import PreTrainedModel
3
+ from .modeling_esm import NTForMaskedLM, MultiHeadAttention
4
+ from .esm_config import NTConfig
5
+ from .modeling_esm_original import EsmForMaskedLM
6
  from transformers.models.esm.configuration_esm import EsmConfig
7
  from enformer_pytorch import Enformer, str_to_one_hot, EnformerConfig
8
  import torch