harrim-nv thhaus commited on
Commit
0e8a28f
·
1 Parent(s): 5d8309d

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>

Files changed (1) hide show
  1. 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,