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
|