Fix import for latest Transformers compatibility
Browse files- Updated `modeling_esm.py` to import `find_pruneable_heads_and_indices` from `transformers.utils.modeling_utils` instead of `transformers.modeling_utils`.
- Ensures compatibility with Transformers >=4.30.
- Enables AutoModelForSequenceClassification to load the model without ImportError.
- modeling_esm.py +2 -5
modeling_esm.py
CHANGED
|
@@ -33,11 +33,8 @@ from transformers.modeling_outputs import (
|
|
| 33 |
SequenceClassifierOutput,
|
| 34 |
TokenClassifierOutput,
|
| 35 |
)
|
| 36 |
-
from transformers.modeling_utils import
|
| 37 |
-
|
| 38 |
-
find_pruneable_heads_and_indices,
|
| 39 |
-
prune_linear_layer,
|
| 40 |
-
)
|
| 41 |
from transformers.utils import logging
|
| 42 |
|
| 43 |
from .esm_config import EsmConfig
|
|
|
|
| 33 |
SequenceClassifierOutput,
|
| 34 |
TokenClassifierOutput,
|
| 35 |
)
|
| 36 |
+
from transformers.modeling_utils import PreTrainedModel
|
| 37 |
+
from transformers.pytorch_utils import find_pruneable_heads_and_indices, prune_linear_layer
|
|
|
|
|
|
|
|
|
|
| 38 |
from transformers.utils import logging
|
| 39 |
|
| 40 |
from .esm_config import EsmConfig
|