Skip to content

GitLab

Unverified Commit 74d7c24d authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Import torch.utils.checkpoint in ProphetNet (#11214)

parent 38a10c6b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/prophetnet/modeling_prophetnet.py
View file @ 74d7c24d
...@@ -22,6 +22,7 @@ from typing import Optional, Tuple ...@@ -22,6 +22,7 @@ from typing import Optional, Tuple
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import torch.utils.checkpoint
from torch import Tensor, nn from torch import Tensor, nn
from torch.nn import LayerNorm from torch.nn import LayerNorm
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment