Update modeling_qformer to support latest transformers (#1)
Browse files- Update modeling_qformer to support latest transformers (ad2605ff65ca47c9b7f88d94bbaa918275226f44)
Co-authored-by: Ao Tang <thhaus@users.noreply.huggingface.co>
- 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,
|