Spaces:
Sleeping
Sleeping
Commit
·
c8da8bd
1
Parent(s):
ee148e0
Remove unused dependency
Browse files- 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,
|
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
|