prithivMLmods commited on
Commit
244823a
·
verified ·
1 Parent(s): e954b70

Update modeling_deepseekocr.py

Browse files
Files changed (1) hide show
  1. 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
- #DeepseekV2MoE,
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