Skip to content

GitLab

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
from typing import Optional, Tuple
import torch
import torch.utils.checkpoint
from torch import nn
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