Update modeling_qheart.py

modeling_qheart.py

@@ -26,11 +26,18 @@ from transformers.models.bert.modeling_bert import (
     BertConfig,
     BertPredictionHeadTransform,
 )
-from transformers.modeling_utils import (
-    apply_chunking_to_forward,
-    find_pruneable_heads_and_indices,
-    prune_linear_layer,
-)
+try:
+  from transformers.pytorch_utils import (
+      apply_chunking_to_forward,
+      find_pruneable_heads_and_indices,
+      prune_linear_layer,
+  )
+except ImportError:
+  from transformers.modeling_utils import (
+      apply_chunking_to_forward,
+      find_pruneable_heads_and_indices,
+      prune_linear_layer,
+  )
 from transformers.activations import ACT2FN
 from peft import LoraConfig, get_peft_model, TaskType