Update modeling_qformer to support latest transformers
#1
by
thhaus - opened
- modeling_qformer.py +2 -0
modeling_qformer.py
CHANGED
|
@@ -41,6 +41,8 @@ from transformers.modeling_outputs import (
|
|
| 41 |
)
|
| 42 |
from transformers.modeling_utils import (
|
| 43 |
PreTrainedModel,
|
|
|
|
|
|
|
| 44 |
apply_chunking_to_forward,
|
| 45 |
find_pruneable_heads_and_indices,
|
| 46 |
prune_linear_layer,
|
|
|
|
| 41 |
)
|
| 42 |
from transformers.modeling_utils import (
|
| 43 |
PreTrainedModel,
|
| 44 |
+
)
|
| 45 |
+
from transformers.pytorch_utils import (
|
| 46 |
apply_chunking_to_forward,
|
| 47 |
find_pruneable_heads_and_indices,
|
| 48 |
prune_linear_layer,
|