[Deepseek] remove redundant code (#5888)

* [misc] fix typos * [Feature] deepseek support via auto model, remove modeling file * [misc] delete useless file * [misc] fix typos * [misc] remove redundant code
hpcaitech · Jul 5, 2024 · d9f683c · d9f683c
1 parent de77e40
commit d9f683c
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/colossalai/shardformer/policies/auto_policy.py b/colossalai/shardformer/policies/auto_policy.py
@@ -167,12 +167,6 @@ class PolicyLocation:
     "transformers_modules.modeling_deepseek.DeepseekForCausalLM": PolicyLocation(
         file_name="deepseek", class_name="DeepseekForCausalLMPolicy"
     ),
-    "colossalai.shardformer.modeling.deepseek_moe_16b_base.modeling_deepseek.DeepseekModel": PolicyLocation(
-        file_name="deepseek", class_name="DeepseekModelPolicy"
-    ),
-    "colossalai.shardformer.modeling.deepseek_moe_16b_base.modeling_deepseek.DeepseekForCausalLM": PolicyLocation(
-        file_name="deepseek", class_name="DeepseekForCausalLMPolicy"
-    ),
     # Falcon
     "transformers.models.falcon.modeling_falcon.FalconModel": PolicyLocation(
         file_name="falcon", class_name="FalconModelPolicy"