Skip to content

GitLab

"docs/source/en/serialization.md" did not exist on "8c14b342aad0ff112a26102019a07bf23d2e33fd"
Unverified Commit bdfcbe60 authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

[Whisper] Fix gradient checkpointing (#19538)

parent 4edb3e49
Show 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
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