Skip to content

GitLab

Unverified Commit bdfcbe60 authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

[Whisper] Fix gradient checkpointing (#19538)

parent 4edb3e49
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/whisper/modeling_whisper.py
View file @ bdfcbe60
...@@ -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