Commit ·
1953026
1
Parent(s): efc1d81
update
Browse files- modeling_protenrich.py +1 -1
modeling_protenrich.py
CHANGED
|
@@ -9,7 +9,7 @@ from transformers import PreTrainedModel
|
|
| 9 |
from transformers.modeling_outputs import BaseModelOutputWithPooling, SequenceClassifierOutput
|
| 10 |
from transformers import AutoModel
|
| 11 |
|
| 12 |
-
from configuration_protenrich import ProtEnrichConfig
|
| 13 |
|
| 14 |
@dataclass
|
| 15 |
class ProtEnrichModelOutput(ModelOutput):
|
|
|
|
| 9 |
from transformers.modeling_outputs import BaseModelOutputWithPooling, SequenceClassifierOutput
|
| 10 |
from transformers import AutoModel
|
| 11 |
|
| 12 |
+
from .configuration_protenrich import ProtEnrichConfig
|
| 13 |
|
| 14 |
@dataclass
|
| 15 |
class ProtEnrichModelOutput(ModelOutput):
|