Skip to content

GitLab

"...en/_static/git@developer.sourcefind.cn:wangsen/mineru.git" did not exist on "fbc7611ee3a73dd587ed7b2f8e1be9e6344742f0"
Unverified Commit 38dfb40a authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

import torch.utils.checkpoint (#15427)

parent f624249d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/xglm/modeling_xglm.py
View file @ 38dfb40a
...@@ -20,6 +20,7 @@ import random ...@@ -20,6 +20,7 @@ import random
from typing import Optional, Tuple from typing import Optional, Tuple
import torch import torch
import torch.utils.checkpoint
from torch import nn from torch import nn
from torch.nn import CrossEntropyLoss from torch.nn import CrossEntropyLoss
......
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