Update modeling_isoformer.py

modeling_isoformer.py

@@ -1,10 +1,8 @@
+from isoformer_config import IsoformerConfig
 from transformers import PreTrainedModel
-#from genomics_research.biobrain_p2.huggingface.modeling_enformer import Enformer
-from genomics_research.biobrain_p2.huggingface.modeling_esm import NTForMaskedLM, MultiHeadAttention
-from genomics_research.biobrain_p2.huggingface.isoformer_config import IsoformerConfig
-#from genomics_research.biobrain_p2.huggingface.enformer_config import EnformerConfig
-from genomics_research.biobrain_p2.huggingface.esm_config import NTConfig
-from genomics_research.biobrain_p2.huggingface.modeling_esm_original import EsmForMaskedLM
+from modeling_esm import NTForMaskedLM, MultiHeadAttention
+from esm_config import NTConfig
+from modeling_esm_original import EsmForMaskedLM
 from transformers.models.esm.configuration_esm import EsmConfig
 from enformer_pytorch import Enformer, str_to_one_hot, EnformerConfig
 import torch