isoformer-anonymous
commited on
Commit
•
13a7e78
1
Parent(s):
0e799f0
Update modeling_isoformer.py
Browse files- 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
|