thhaus commited on
Commit
d87e489
·
verified ·
1 Parent(s): 9f4ff4d

Update modeling_qformer.py

Browse files

Update modeling_qformer to support latest transformers

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,