lmb-freiburg
diff --git a/‎ovqa/models/lavis/blip2_models/Qformer.py‎
Lines changed: 2 additions & 3 deletions b/‎ovqa/models/lavis/blip2_models/Qformer.py‎
Lines changed: 2 additions & 3 deletions
@@ -20,11 +20,10 @@
     CausalLMOutputWithCrossAttentions,
     MaskedLMOutput,
 )
+from transformers import apply_chunking_to_forward
+from transformers.pytorch_utils import find_pruneable_heads_and_indices, prune_linear_layer
 from transformers.modeling_utils import (
     PreTrainedModel,
-    apply_chunking_to_forward,
-    find_pruneable_heads_and_indices,
-    prune_linear_layer,
 )
 from transformers.models.bert.configuration_bert import BertConfig
 from transformers.utils import logging
Original file line number	Diff line number	Diff line change
`@@ -20,11 +20,10 @@`
`20`	`20`	`CausalLMOutputWithCrossAttentions,`
`21`	`21`	`MaskedLMOutput,`
`22`	`22`	`)`
	`23`	`+from transformers import apply_chunking_to_forward`
	`24`	`+from transformers.pytorch_utils import find_pruneable_heads_and_indices, prune_linear_layer`
`23`	`25`	`from transformers.modeling_utils import (`
`24`	`26`	`PreTrainedModel,`
`25`		`- apply_chunking_to_forward,`
`26`		`- find_pruneable_heads_and_indices,`
`27`		`- prune_linear_layer,`
`28`	`27`	`)`
`29`	`28`	`from transformers.models.bert.configuration_bert import BertConfig`
`30`	`29`	`from transformers.utils import logging`