Fix: torch.utils.checkpoint import error. (#17849)

src/transformers/models/deberta/modeling_deberta.py

@@ -19,6 +19,7 @@ from collections.abc import Sequence
 from typing import Optional, Tuple, Union
 
 import torch
+import torch.utils.checkpoint
 from torch import nn
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 

src/transformers/models/deberta_v2/modeling_deberta_v2.py

@@ -20,6 +20,7 @@ from typing import Optional, Tuple, Union
 
 import numpy as np
 import torch
+import torch.utils.checkpoint
 from torch import nn
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, LayerNorm, MSELoss