ejschwartz commited on
Commit
c8da8bd
·
1 Parent(s): ee148e0

Remove unused dependency

Browse files
Files changed (1) hide show
  1. modeling_nova.py +1 -1
modeling_nova.py CHANGED
@@ -5,7 +5,7 @@ import torch.nn as nn
5
  import torch.nn.functional as F
6
  from typing import Tuple, List, Optional
7
  from transformers import LlamaModel, LlamaConfig, LlamaForCausalLM
8
- from transformers.models.llama.modeling_llama import LlamaDecoderLayer, LLAMA_ATTENTION_CLASSES, LlamaMLP, LlamaRMSNorm
9
  from transformers.models.llama.modeling_llama import LlamaSdpaAttention, apply_rotary_pos_emb, repeat_kv
10
  from transformers import logging, Cache, DynamicCache, StaticCache
11
  from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast
 
5
  import torch.nn.functional as F
6
  from typing import Tuple, List, Optional
7
  from transformers import LlamaModel, LlamaConfig, LlamaForCausalLM
8
+ from transformers.models.llama.modeling_llama import LlamaDecoderLayer, LlamaMLP, LlamaRMSNorm
9
  from transformers.models.llama.modeling_llama import LlamaSdpaAttention, apply_rotary_pos_emb, repeat_kv
10
  from transformers import logging, Cache, DynamicCache, StaticCache
11
  from transformers.modeling_outputs import BaseModelOutputWithPast, CausalLMOutputWithPast