From d7355db6ae0c73fb269185f74bd799bf6615508e Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 30 May 2024 16:48:48 +0200 Subject: [PATCH] nit --- src/transformers/models/gemma/diff_gemma.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/models/gemma/diff_gemma.py b/src/transformers/models/gemma/diff_gemma.py index c4be5e3e20dd67..6068dff45c0d85 100644 --- a/src/transformers/models/gemma/diff_gemma.py +++ b/src/transformers/models/gemma/diff_gemma.py @@ -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,