Skip to content

Commit

Permalink
nit
Browse files Browse the repository at this point in the history
  • Loading branch information
ArthurZucker committed May 30, 2024
1 parent e1b0262 commit d7355db
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/transformers/models/gemma/diff_gemma.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
import torch.utils.checkpoint
from torch import nn
from torch.nn import CrossEntropyLoss

from transformers import PretrainedConfig
from transformers.models.llama.modeling_llama import (
LlamaForCausalLM,
Expand Down

0 comments on commit d7355db

Please sign in to comment.