harrim-nv thhaus commited on
Commit
787e0b9
·
1 Parent(s): 85f5627

Update modeling_qformer to support latest transformers (#1)

Browse files

- Update modeling_qformer to support latest transformers (037dac6503791eab445d05542e0f5c5a478e911c)


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,