mila-intel
/

ProtST-esm1b

Feature Extraction

Model card Files Files and versions Community

Jiqing commited on Jan 2, 2024

Commit

401c15c

·

1 Parent(s): ea455b4

Update modeling_protst.py

Files changed (1) hide show

modeling_protst.py +1 -1

modeling_protst.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import PreTrainedModel
 from transformers.modeling_outputs import ModelOutput
 from transformers.models.esm import EsmPreTrainedModel, EsmModel
 from transformers.models.bert import BertPreTrainedModel, BertModel
-from configuration_protst import ProtSTConfig
 @dataclass

 from transformers.modeling_outputs import ModelOutput
 from transformers.models.esm import EsmPreTrainedModel, EsmModel
 from transformers.models.bert import BertPreTrainedModel, BertModel
+from .configuration_protst import ProtSTConfig
 @dataclass