Update modeling_deepseekocr.py
Browse files- modeling_deepseekocr.py +1 -1
modeling_deepseekocr.py
CHANGED
|
@@ -20,7 +20,7 @@ from transformers import DeepseekV2Model, DeepseekV2ForCausalLM
|
|
| 20 |
from transformers import DeepseekV2Config
|
| 21 |
from transformers.models.deepseek_v2.modeling_deepseek_v2 import (
|
| 22 |
DeepseekV2Attention, DeepseekV2MLP,
|
| 23 |
-
|
| 24 |
DeepseekV2RMSNorm, DeepseekV2DecoderLayer)
|
| 25 |
from transformers.models.llama.modeling_llama import LlamaAttention, LlamaRotaryEmbedding
|
| 26 |
from transformers import TextStreamer
|
|
|
|
| 20 |
from transformers import DeepseekV2Config
|
| 21 |
from transformers.models.deepseek_v2.modeling_deepseek_v2 import (
|
| 22 |
DeepseekV2Attention, DeepseekV2MLP,
|
| 23 |
+
DeepseekV2MoE,
|
| 24 |
DeepseekV2RMSNorm, DeepseekV2DecoderLayer)
|
| 25 |
from transformers.models.llama.modeling_llama import LlamaAttention, LlamaRotaryEmbedding
|
| 26 |
from transformers import TextStreamer
|