Update modeling_qformer to support latest transformers

#1
by thhaus - opened
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,