style

huggingface · Dec 17, 2024 · 7bdf61c · 7bdf61c
1 parent 5b4ebaa
commit 7bdf61c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/models/granite/modular_granite.py b/src/transformers/models/granite/modular_granite.py
@@ -17,18 +17,18 @@
 
 import torch
 import torch.utils.checkpoint
-from torch import nn
 
 from ...cache_utils import Cache
 from ...utils import (
     logging,
 )
-from ..llama.modeling_llama import LlamaAttention, LlamaForCausalLM, LlamaDecoderLayer
+from ..llama.modeling_llama import LlamaAttention, LlamaDecoderLayer, LlamaForCausalLM
 from .configuration_granite import GraniteConfig
 
 
 logger = logging.get_logger(__name__)
 
+
 class GraniteAttention(LlamaAttention):
     """Multi-headed attention from 'Attention Is All You Need' paper"""